optimizer
Functions
Moves the state of a single optimizer to the device. |
|
Moves optimizer states for a sequence of optimizers to the device. |
- pytorch_lightning.utilities.optimizer.optimizer_to_device(optimizer, device)[source]
Moves the state of a single optimizer to the device.
- Return type