Update README.md

roatienza · Oct 21, 2021 · 37d039e · 37d039e
1 parent b240408
commit 37d039e
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -37,3 +37,26 @@ Soon.
 ```
 pip3 install -r requirements.txt
 ```
+
+### Train
+
+For example, train ResNet50 with Agmax on 2 GPUs for 90 epochs, SGD with `lr=0.1` and multistep learning rate:
+
+```
+CUDA_VISIBLE_DEVICES=0,1 python3 main.py --config=ResNet50-standard-agmax --train --multisteplr \
+--dataset=imagenet --epochs=90 --save
+```
+
+Compare the results without Agmax:
+
+```
+CUDA_VISIBLE_DEVICES=0,1 python3 main.py --config=ResNet50-standard --train --multisteplr \
+--dataset=imagenet --epochs=90 --save
+```
+
+### Test
+
+
+```
+```
+