Update README.md

dumpmemory · Jul 12, 2022 · 30a92d0 · 30a92d0
1 parent 9c34d80
commit 30a92d0
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -5,10 +5,20 @@
 Fei Zhu, Xu-Yao Zhang, Chuang Wang,  Fei Yin, Cheng-Lin Liu<br>
 [Paper](https://openaccess.thecvf.com/content/CVPR2021/papers/Zhu_Prototype_Augmentation_and_Self-Supervision_for_Incremental_Learning_CVPR_2021_paper.pdf)
 ### Usage 
-We run the code with torch version:1.10.0, python version:3.9.7
+We run the code with torch version: 1.10.0, python version: 3.9.7
 * Train CIFAR100
 ```
-run `main.py`.
+python main.py
+```
+* Train Tiny-ImageNet
+```
+cd Tiny-ImageNet
+python main_tiny.py
+```
+* Train ImageNet-Subset
+```
+cd ImageNet-Subset
+python main_PASS_imagenet.py
 ```
 
 ### Citation