Skip to content

Instantly share code, notes, and snippets.

@tinhb92

tinhb92/train_search_learner.py

Created April 17, 2019 16:58
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save tinhb92/66f53b99f577d8d06af4343de5798e50 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save tinhb92/66f53b99f577d8d06af4343de5798e50 to your computer and use it in GitHub Desktop.
Download ZIP
train search learner
Raw
train_search_learner.py
learn = Learner(dat, DartsRnnSearch(),
opt_func = torch.optim.SGD,
callback_fns = [
HidInit,
partial(ArchParamUpdate, search_dat=search_dat,
arch_lr=arch_lr, arch_wdecay=arch_wdecay, wdecay=wdecay),
Regu,
PrintGenotype,
partial(GradientClipping, clip=clip),
partial(CSVLogger, filename = csv_name, append = True),
GcCol,
# partial(ResumeModel, name = resume_model)
],
wd = wdecay
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment