Skip to content

Commit 2fd3396

Browse files
authored
Update README.md
1 parent d7f5b51 commit 2fd3396

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

README.md

+1
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ For example, to train the 1-shot 5-way FEAT model with ConvNet backbone on MiniI
7777
$ python train_feat.py --lr 0.0001 --temperature 32 --max_epoch 200 --model_type ConvNet --dataset MiniImageNet --init_weights ./saves/initialization/miniimagenet/con-pre.pth --shot 1 --way 5 --gpu 0 --balance 10 --step_size 50 --gamma 0.1 --lr_mul 10
7878

7979
to train the 1-shot 5-way FEAT model with ResNet backbone on MiniImageNet:
80+
8081
$ python train_feat.py --lr 0.0001 --temperature 128 --max_epoch 100 --model_type ResNet --dataset MiniImageNet --init_weights ./saves/initialization/miniimagenet/res-pre.pth --shot 1 --way 5 --gpu 0 --balance 10 --step_size 10 --gamma 0.5 --lr_mul 10
8182

8283

0 commit comments

Comments
 (0)