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);
|
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);
|