`mle init` and many small improvements - Githubissues

mle-infrastructure / mle-toolbox

Lightweight Tool to Manage Distributed ML Experiments 🛠

https://mle-infrastructure.github.io/mle_toolbox/toolbox/

MIT License

3 stars 1 forks source link

`mle init` and many small improvements #48

Closed RobertTLange closed 3 years ago

RobertTLange commented 3 years ago

Adresses #24 and adds a subcommand mle init. Furthermore there are a couple of small bug fixes, a lot of refactoring and new helper functions:

Fix report generation bug (everything has to be a string for markdown-ification!).
Fix monitor bug: No longer reload the local database at each update call.
Adds get_jax_os_ready helper for setting up JAX environment variables.
Adds load_model_ckpt for smooth reloading of stored checkpoints.