Skip to content

Deep Learning Framework for Reproducible Experiments

License

Notifications You must be signed in to change notification settings

MIPT-Oulu/Collagen

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Collagen: Deep Learning Framework for Reproducible Experiments

Build Status

About

Install

You can get the most stable version from pip:

pip install collagen

The fresh-most release of the framework can also be installed from the master branch.

Collagen supports distributed training with mixed precision out-of-the box. Consider using APEX to use this feature:

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext"