research project
Project description
OMLET: Open Machine Learning Engineering Toolbox
Installation
git clone https://github.com/LinxiFan/omlet.git
cd omlet
pip install -e .
PyTorch-Lightning
Tested only with v0.7.6, please upgrade.
Hydra
Due to incompatibilities, please install hydra
management library from their master branch:
git clone https://github.com/facebookresearch/hydra.git
cd hydra
pip install -e .
cd plugins/hydra_colorlog
pip install -e .
The above should also install OmegaConf 2.0.1, which we'll use as our primary dot-access dict for hyperparameters.
OmegaConf docs: https://omegaconf.readthedocs.io/en/latest/usage.html
Nvidia APEX fp16 training
https://github.com/NVIDIA/apex#quick-start
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
Project details
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
omlet-0.1.tar.gz
(32.7 kB
view hashes)
Built Distribution
omlet-0.1-py3-none-any.whl
(41.2 kB
view hashes)