Hi
In your lightning module, you could do this:
def on_epoch_start(self):
if self.loss_SUM > 0.3
self.trainer.optimizers[0] = Adam(...)
and you start with LBFGS as default, returned in configure_optimizers.
Hi
In your lightning module, you could do this:
def on_epoch_start(self):
if self.loss_SUM > 0.3
self.trainer.optimizers[0] = Adam(...)
and you start with LBFGS as default, returned in configure_optimizers.