HandyRL icon indicating copy to clipboard operation
HandyRL copied to clipboard

(Idea) feature: set optimizer from outside

Open YuriCat opened this issue 4 years ago • 1 comments

example

net = SimpleConv2dModel2()
optim = Adam(net.parameters(), lr=1e-4)

learner = Learner(args=args, net=net, optim=optim, remote=False)
learner.run()

This PR requires #170 (return model instance by calling net())

YuriCat avatar Apr 29 '21 19:04 YuriCat

I like this idea because the optimizer often affects the performance.

ikki407 avatar May 06 '21 04:05 ikki407