Skip to content

Loss_balancing as a LightningModule attribute

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)

Merge request reports