diff --git a/research_seed/mnist/README.md b/research_seed/mnist/README.md index 2667224178e6f11f1c49f6b84a5d6dbd3125610f..edb17e979e73d51a7e4692f13755914b14e81240 100644 --- a/research_seed/mnist/README.md +++ b/research_seed/mnist/README.md @@ -3,15 +3,15 @@ In this readme, give instructions on how to run your code. #### CPU ```python -python lightning_modules/train.py +python mnist_train.py ``` #### Multiple-GPUs ```python -python lightning_modules/train.py --gpus '0,1,2,3' +python mnist_train.py --gpus '0,1,2,3' ``` #### On multiple nodes ```python -python lightning_modules/train.py --gpus '0,1,2,3' --nodes 4 +python mnist_train.py --gpus '0,1,2,3' --nodes 4 ```