automl / NASLib

NASLib is a Neural Architecture Search (NAS) library for facilitating NAS research for the community by providing interfaces to several state-of-the-art NAS search spaces and optimizers.
Apache License 2.0
513 stars 117 forks source link

Fix DartsSearchSpace, Trainer.evaluate, and update tests #83

Closed Neonkraft closed 2 years ago

Neonkraft commented 2 years ago

This PR does three things:

  1. Removes extraneous non-affine batchnorms present in the DartsSearchSpace graph
  2. Fixes the validation code in Trainer.evaluate
  3. Updates tests

The end result is that the darts search space validation curves now resemble the original implementation.