mnist_stages

Catalyst: MNIST with stages example

Local run

From the examples folder:

# 3 options available
bash mnist_stages/run_config.sh  # for Config API run
bash mnist_stages/run_hydra.sh   # for Hydra API run
bash mnist_stages/run_tune.sh    # for Tune run 

Training visualization

For tensorboard visualization use

CUDA_VISIBLE_DEVICE="" tensorboard --logdir=./logs

To change number of epochs for train or tune stages you can use command-line

catalyst-dl run --config config.yml --stages/train/num_epochs=10:int --stages/tune/num_epochs=10:int