/experii

An experimentation framework. Integrating VertaAI's ModelDB for versioning and Ax platform for hyperparameters optimization

Primary LanguagePython

experii

An experimentation framework. Integrating VertaAI's ModelDB for model versioning and Ax platform for hyperparameters optimization with my nn-toolbox.

Current Functionality

  • Logging hyperparameters, model specifications, checkpoints, etc. to ModelDB
  • Hyperparameter optimization via Ax platform's loop API.

Planned Functionality

  • Integration with more hyperparameter optimization frameworks

Example

Take a look at test_ax.py for an example on tuning SGD's learning rates.