File size: 262 Bytes
cc9c7ee
 
 
 
 
 
 
1
2
3
4
5
6
7
8
" Method containing activation functions"
from torch.optim import Adam, AdamW, SGD
from src.utils.mapper import configmapper

configmapper.map("optimizers", "adam")(Adam)
configmapper.map("optimizers", "adam_w")(AdamW)
configmapper.map("optimizers", "sgd")(SGD)