ML logging

Logging using aimstack from https://aimstack.io/, and pytorch lightening.

Code

Once work done and tested,

1. bump version
2. merge the code to the main branch

Then it will be auto-pushed to the pypi repository.

Use

pip install --extra-index-url https://pypi.fury.io/kdjfaosdjfaisd/ torch_wrapper

ML python environment