/pytorch-project-framework

A simple and custom PyTorch project template.

Primary LanguagePython

This is a custom PyTorch Project Template to make the projects quick and easy to build.

  • Things to add:
    • Hyperparameter Tuning (preferably with Optuna)
    • Additional utilities in API