optimizer (Literal
['Adam'
, 'SGD'
]) – The name of the optimizer to be used. Options: Adam
or SGD
.
lr (float
) – Learning rate to be used.
momentum (float
) – Value for the momentum hyperparameter (if relevant).
weight_decay (float
) – Value for the weight_decay hyperparameter (if relevant).