examples

Catalyst examples

Tutorials

1. classification tutorial
   • dataset preparation (raw images -> train/valid/infer splits)
   • augmentations usage example
   • pretrained model finetuning
   • various classification metrics
   • metrics visualizaiton
   • FocalLoss and OneCycle usage examples
   • class imbalance handling
   • model inference
2. segmentation tutorial
   • car segmentation dataset
   • augmentations with albumentations library
   • training in FP16 with NVIDIA Apex
   • using segmentation models from catalyst/contrib/models/segmentation
   • training with multiple criterion (Dice + IoU + BCE) example
   • Lookahead + RAdam optimizer usage example
   • tensorboard logs visualization
   • predictions visualization
   • Test-time augmentations with ttach library

Usage examples

1. features – classification
   • cifar10 classification model
   • Runner usage example
2. features – segmentation
   • segmentation with unet
   • model training and inference
   • predictions visialization

---

Pipelines

Full description of configs with comments:

• Eng
• Rus

1. classification pipeline
   • classification model training and inference
   • different augmentations and stages usage
   • metrics visualization with tensorboard
2. segmentation pipeline
   • binary and semantic segmentation with U-Net
   • model training and inference
   • different augmentations and stages usage
   • metrics visualization with tensorboard

Usage examples

1. features – model training
   • configuration files usage example
   • local and docker runs
   • metrics visualization with tensorboard
2. features – model training with stages
   • pipeline example with stages
3. features - vanilla GAN on MNIST
   • experiment with multiple phases & models & optimizers
4. features - Text classification with DistilBert
   • DistilBERT model
   • Kaggle kernel with the same example

---

RL tutorials & pipelines

For Reinforcement Learning examples check out our Catalyst.RL repo

---

Catalyst-info

Link

Contributions

We supervise the Awesome Catalyst list. You can make a PR with your project to the list.