Update README

moyix · Nov 27, 2020 · 7941beb · 7941beb
1 parent 7ad1b9c
commit 7941beb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -9,10 +9,10 @@ I'm playing with [PyTorch](http://pytorch.org/) on the CIFAR10 dataset.
 ## Training
 ```
 # Start training with: 
-CUDA_VISIBLE_DEVICES=0 python main.py
+python main.py
 
 # You can manually resume the training with: 
-CUDA_VISIBLE_DEVICES=0 python main.py --resume --lr=0.01
+python main.py --resume --lr=0.01
 ```
 
 ## Accuracy