Skip to content
GitLab

Loss_balancing as a LightningModule attribute

Merged Mikael Jacquemont requested to merge logvar_ckpt into master

Add loss_balancing as a lightningModule attribute to allow for a correct device setting and checkpoint of loss_balancing parameters.

fixes #103 (closed)