We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d7f5b51 commit 2fd3396Copy full SHA for 2fd3396
README.md
@@ -77,6 +77,7 @@ For example, to train the 1-shot 5-way FEAT model with ConvNet backbone on MiniI
77
$ 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
78
79
to train the 1-shot 5-way FEAT model with ResNet backbone on MiniImageNet:
80
+
81
$ 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
82
83
0 commit comments