MLgym, a python framework for distributed machine learning model training in research.
Project description
MLgym
a python framework for distributed machine learning model training in research.
At its core, MLgym offers functionality to run gridsearches of Pytorch models at scale split over multiple GPUs and centrally store the results using DashifyML.
Futhermore, MLgym provides the following key features:
-
Reproducibility of results due to full experiment specification including dataset, preprocessing routines, model architecture, loss function, metrics and more within a single YAML config.
-
Component registry to register custom components with dependencies. For instance one can define a new preprocessing routine component. This component may depend on an iterator component, as specified in the experiment config. During runtime these components are instantiated on the fly.
-
Resume training after crash
-
Custom training routines, e.g., training with partially frozen network weights
-
Large scale, multi GPU training supporting Grid Search, Nested Cross Validation, Cross Validation
-
Reduced logging to reduce storage footprint of model and optimizer states
Please note, that at the moment this code should be treated as experimental and is not production ready.
Install
there are two options to install MLgym, the easiest way is to install it from the pip repository:
pip install mlgym
For the latest version, one can directly install it from source by cd into the root folder and then running
pip install src/
Usage
NOTE: This framework is still under heavy development and mainly used in research projects. It's most likely not free of bugs and interfaces can still change.
For usage see this example.
Copyright
Copyright (c) 2020 Max Lübbering, Rajkumar Ramamurthy
For license see: https://github.com/le1nux/mlgym/blob/master/LICENSE
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file mlgym-0.0.58.tar.gz.
File metadata
- Download URL: mlgym-0.0.58.tar.gz
- Upload date:
- Size: 44.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.0 CPython/3.9.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | f44256caad0a246816368b8f20438f3312e2e54c9d774cb7f340a2f2ac2158db |
|
| MD5 | d764c509e30d0fecc829ecab5d9e5d65 |
|
| BLAKE2b-256 | 37fe85937b4d0b7753447c8df8c1b88d18b02418a0e5bae553c0321379b11bc6 |
File details
Details for the file mlgym-0.0.58-py3-none-any.whl.
File metadata
- Download URL: mlgym-0.0.58-py3-none-any.whl
- Upload date:
- Size: 63.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.0 CPython/3.9.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | f4e17892a6c518e542c1d1a9aa19dc22c4dfacbe916ccc0e232c3993b70ef902 |
|
| MD5 | 482a8c2de221f74b6471ca1a5e51fe68 |
|
| BLAKE2b-256 | 8e635403a1e8682f2f1245c25e3de4febe0ca90c9eb93dead7d31d4ed7363e83 |
