-
Notifications
You must be signed in to change notification settings - Fork 0
/
reproduce.sh
33 lines (31 loc) · 1.39 KB
/
reproduce.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#!/usr/bin/env bash
CUDA_VISIBLE_DEVICES=0 python FT_cifar.py \
--gpu 0 \
--dataset CIFAR100 \
--way 5 \
--shot 1 \
--lr 0.001 \
--step_size 50 \
--batch_size 50 \
--TASK_NUM 16 \
--BASE_CLS_NUM 5 \
--gamma 0.8 \
--model convnet \
--dim 1600 \
--method ft \
--max_epoch 200
CUDA_VISIBLE_DEVICES=0 python ERD_cifar.py \
--gpu 0 \
--dataset CIFAR100 \
--way 5 \
--shot 1 \
--lr 0.001 \
--step_size 50 \
--batch_size 50 \
--TASK_NUM 16 \
--BASE_CLS_NUM 5 \
--gamma 0.8 \
--model convnet \
--dim 1600 \
--method ft \
--max_epoch 200