File size: 188 Bytes
430712c |
1 2 3 4 5 6 7 8 9 10 11 |
#! /usr/bin/python
# -*- encoding: utf-8 -*-
import torch
def Optimizer(parameters, lr, **kwargs):
print('Initialised Adam optimizer')
return torch.optim.AdamW(parameters, lr = lr);
|