Shortcuts

pytorch_lightning.trainer.optimizers module

class pytorch_lightning.trainer.optimizers.TrainerOptimizersMixin[source]

Bases: abc.ABC

configure_schedulers(schedulers)[source]
init_optimizers(model)[source]
Return type

Tuple[List, List, List]

reinit_scheduler_properties(optimizers, schedulers)[source]
class pytorch_lightning.trainer.optimizers._MockOptimizer[source]

Bases: torch.optim.optimizer.Optimizer

The _MockOptimizer will be used inplace of an optimizer in the event that None is returned from configure_optimizers.

add_param_group(param_group)[source]
load_state_dict(state_dict)[source]
state_dict()[source]
step(closure=None)[source]
zero_grad()[source]