Skip to content

Latest commit

 

History

History
27 lines (23 loc) · 867 Bytes

README.md

File metadata and controls

27 lines (23 loc) · 867 Bytes

catalyst_example

an example of the training file supporting distributed training and curriculum learning with Catalyst

installation

  1. Create and populate the environment
MYNEWENV="" # write a name of your environment in quotes, like "torch"
conda create --name ${MYNEWENV} python=3.9
conda activate ${MYNEWENV}
pip3 install torch torchvision torchaudio
pip3 install -U catalyst
conda install -c anaconda nccl
pip install  mindfultensors
pip3 install pymongo
pip3 install nibabel
pip3 install pynvml
pip3 install scipy
  1. Use the code in this repository to train your model
    1. The main file is curriculum_training.py
    2. Edit everything.
    3. Do not forget to set WANDBTEAM to the value of your team, so your logs are in the correct place
    4. Mainly make sure that get_model method of the CustomRunner class initializes your model