Source code for mmaction.core.optimizer.copy_of_sgd

from mmcv.runner import OPTIMIZERS
from torch.optim import SGD


[docs]@OPTIMIZERS.register_module() class CopyOfSGD(SGD): """A clone of torch.optim.SGD. A customized optimizer could be defined like CopyOfSGD. You may derive from built-in optimizers in torch.optim, or directly implement a new optimizer. """