update

CHENGY12 · Jan 24, 2023 · c8d1a35 · c8d1a35
1 parent d10c8b1
commit c8d1a35
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/plot-coop/README.md b/plot-coop/README.md
@@ -15,7 +15,6 @@ Please follow the instructions [DATASETS.md](https://github.com/KaiyangZhou/CoOp
 
 
 The running scripts are in `scripts/`. `cd ./scripts` and change the `your_data_path` and `your_work_path` in `scripts/main.sh`
-
 Then, you can run the commands `bash main.sh DATASET N` under `CoOp/scripts/`.
 
 `DATASET` takes as input a dataset name, like `caltech101`. 

diff --git a/plot-coop/configs/trainers/PLOT/rn50.yaml b/plot-coop/configs/trainers/PLOT/rn50.yaml
@@ -1,6 +1,6 @@
 DATALOADER:
   TRAIN_X:
-    BATCH_SIZE: 128 #32 for small dataset such as FGVC_air
+    BATCH_SIZE: 128 
   TEST:
     BATCH_SIZE: 200
   NUM_WORKERS: 8

diff --git a/plot-coop/train.py b/plot-coop/train.py
@@ -124,6 +124,9 @@ def setup_cfg(args):
     if cfg.DATASET.NAME== "ImageNet":
         cfg.OPTIM.MAX_EPOCH=50
 
+    if cfg.DATASET.NAME in ['OxfordFlowers','FGVCAircraft','StanfordCars']:
+        cfg.DATALOADER.TRAIN_X.BATCH_SIZE=32  #32 for small dataset such as Car,Air,Flowers
+
     cfg.freeze()
Original file line number	Diff line number	Diff line change
Expand Up		@@ -15,7 +15,6 @@ Please follow the instructions [DATASETS.md](https://github.com/KaiyangZhou/CoOp


		The running scripts are in `scripts/`. `cd ./scripts` and change the `your_data_path` and `your_work_path` in `scripts/main.sh`

		Then, you can run the commands `bash main.sh DATASET N` under `CoOp/scripts/`.

		`DATASET` takes as input a dataset name, like `caltech101`.
Expand Down