Shortcuts

optimizer

Functions

optimizer_to_device

Moves the state of a single optimizer to the device.

optimizers_to_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:

None

pytorch_lightning.utilities.optimizer.optimizers_to_device(optimizers, device)[source]

Moves optimizer states for a sequence of optimizers to the device.

Return type:

None