vm.torch.optim

Classes

class Adam : The optimizer to apply Adam algorithm. [Kingma & Ba, 2014].

class AdamW : The optimizer to apply AdamW algorithm. [Loshchilov & Hutter, 2017].

class LARS : The optimizer to apply LARS algorithm. [You et.al, 2017].

class Optimizer : The base class of optimizers.

class RMSprop : The optimizer to apply RMSprop algorithm. [Hinton et.al, 2013].

class SGD : The optimizer to apply SGD algorithm.