forked from wyharveychen/CloserLookFewShot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
universal_settings.py
33 lines (28 loc) · 1.06 KB
/
universal_settings.py
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
n_base_classes = {
'omniglot':4112, 'omniglot_400cl':400, 'omniglot_40cl':40, # val:688, novel:1692
'CUB':100, 'CUB_base50cl':50, 'CUB_base25cl':25, # val:50, novel:50
'miniImagenet':64, # val:16, novel:20
'cross_char':1597, 'cross_char_half':758, 'cross_char_quarter':350, # val:31, novel:31
'cross_char_base12lang':350, 'cross_char_base3lang':69, 'cross_char_base1lang':20, # val:31, novel:31
'cross_char2':1597, 'cross_char2_quarter':350, 'cross_char2_base3lang':69, 'cross_char2_base1lang':20, # val:31, novel:31
'cross':100, 'cross_base80cl':80, 'cross_base40cl':40, 'cross_base20cl':20, # val:50, novel:50
}
stop_epoch = {
'baseline':500,
'protonet':1000,
# 'relationnet':1000,
'relationnet_softmax':1000,
'matchingnet':1000,
}
patience = {
'baseline':10,
'protonet':70,
# 'relationnet':70,
'relationnet_softmax':70,
'matchingnet':70,
}
should_aug_datasets = [
'CUB', 'CUB_base50cl', 'CUB_base25cl',
'miniImagenet',
'cross', 'cross_base80cl', 'cross_base40cl', 'cross_base20cl'
]