carpedm20 / ENAS-pytorch

PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"
Apache License 2.0
2.69k stars 492 forks source link

Adds a "single" mode that loads and trains a given dag #35

Closed nkcr closed 5 years ago

nkcr commented 5 years ago

When running with --mode single, it loads the dag given by --dag_path (which is a simple json dump) and trains this dag.

In this mode, there is no "autoML" part. The controller is not trained and the same dag is used during the "train_shared" phase and validate phase.

I provided a dag.json file, which is the best dag found in the paper. This mode should allow us to "retrain from scratch" and eventually reach the result of the paper (working on it...).

Major changes are in the Trainer class. I added a single=False optional parameter on the train() method and a dag=None optional parameter on the train_shared() method. This pull does not break backward compatibility.

carpedm20 commented 5 years ago

Super cool! Thanks for the PR.