Skip to content

(Idea) feature: set optimizer from outside#193

Open
YuriCat wants to merge 8 commits intoDeNA:developfrom
YuriCat:feature/set_optimizer_from_outside
Open

(Idea) feature: set optimizer from outside#193
YuriCat wants to merge 8 commits intoDeNA:developfrom
YuriCat:feature/set_optimizer_from_outside

Conversation

@YuriCat
Copy link
Copy Markdown
Contributor

@YuriCat YuriCat commented Apr 29, 2021

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 YuriCat changed the title feature: set optimizer from outside (Idea) feature: set optimizer from outside Apr 29, 2021
@ikki407
Copy link
Copy Markdown
Member

ikki407 commented May 6, 2021

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants