sgd.py 242 B

12345678910
  1. #! /usr/bin/python
  2. # -*- encoding: utf-8 -*-
  3. import torch
  4. def Optimizer(parameters, lr, weight_decay, **kwargs):
  5. print('Initialised SGD optimizer')
  6. return torch.optim.SGD(parameters, lr = lr, momentum = 0.9, weight_decay=weight_decay);