Add Aim integration to PyTorch image classification template - Githubissues

jrieke / traingenerator

🧙 A web app to generate template code for machine learning

https://traingenerator.jrieke.com

MIT License

1.36k stars 181 forks source link

Add Aim integration to PyTorch image classification template #12

Closed gorarakelyan closed 3 years ago

gorarakelyan commented 3 years ago

Summary

Added Aim to PyTorch image classification template as an experiment logger option.

Details

What was added:

Initialization of aim.Session
Logging parameters with Aim
Logging metrics with Aim
Aim as a logger option in the sidebar
Experiment name input in the sidebar
Markdown of how to run UI and link to the full documentation

Checklist

[x] all tests are passing (see README.md on how to run tests)
[x] you formatted all code with black
[x] you checked all new functionality live, i.e. in the running web app
[x] any generated code is formatted nicely, both in .py and in .ipynb ("nicely" = comparable to the existing templates)
[x] you added comments in your code that explain what it does
[x] the PR explains in detail what's new