def get_opt(name):
opts = {
'SGD': optim.SGD,
'Adam': optim.Adam,
'Adagrad': optim.Adagrad,
'RMSprop': optim.RMSprop,
}
return opts[name]
微信
公众号
扫码关注公众号