adam.py 230 B

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