forked from RL-VIG/LibFewShot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathCAN-miniImageNet--ravi-resnet18-5-1-Table2.yaml
71 lines (71 loc) · 1.14 KB
/
CAN-miniImageNet--ravi-resnet18-5-1-Table2.yaml
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
augment: true
augment_times: 1
augment_times_query: 1
backbone:
kwargs:
avg_pool: false
is_feature: false
is_flatten: false
name: resnet18
batch_size: 128
classifier:
kwargs:
HW: 11
nFeat: 512
num_class: 64
scale_cls: 7
name: CAN
data_root: /data/wzy/miniImageNet--ravi
deterministic: true
device_ids: 0,1
episode_size: 8
epoch: 100
image_size: 84
includes:
- headers/data.yaml
- headers/device.yaml
- headers/misc.yaml
- headers/model.yaml
- headers/optimizer.yaml
- classifiers/CAN.yaml
- backbones/Conv64F.yaml
log_interval: 100
log_level: info
log_name: null
log_paramerter: false
lr_scheduler:
kwargs:
T_max: 100
eta_min: 0
name: CosineAnnealingLR
n_gpu: 2
optimizer:
kwargs:
lr: 0.1
momentum: 0.9
nesterov: true
weight_decay: 0.0005
name: SGD
other: null
parallel_part:
- emb_func
pretrain_path: null
query_num: 15
rank: 0
result_root: ./results
resume: false
save_interval: 10
save_part:
- emb_func
seed: 2147483647
shot_num: 1
tag: null
tb_scale: 3.3333333333333335
test_episode: 600
test_epoch: 5
test_query: 15
test_shot: 1
test_way: 5
train_episode: 2000
use_memory: false
way_num: 5