Skip to content

(Idea) feature: set optimizer from outside#193

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

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

Conversation

@YuriCat

@YuriCat YuriCat commented Apr 29, 2021

Copy link
Copy Markdown
Contributor

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

ikki407 commented May 6, 2021

Copy link
Copy Markdown
Member

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