forked from ChenWWWeixiang/diagnosis_covid19
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest.toml
58 lines (51 loc) · 881 Bytes
/
test.toml
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
title = "covid19 PyTorch test"
[general]
usecudnn = true
usecudnnbenchmark = true
gpuid = "6"
loadpretrainedmodel = true
pretrainedmodelpath = 'weights/weak_super.pt'
random_seed = true
use_3d = false
use_lstm=false
use_slice=true
use_plus=false
use25d=false
class_num=4
mod='xct'
plus_as_input=false
[input]
batchsize = 32
numworkers = 8
shuffle = true
[model]
type = "Finetune-label"
inputdim = 256
hiddendim = 256
numclasses = 2
numlstms = 2
[training]
train = true
epochs = 50
startepoch = 0
statsfrequency = 1000
data_root = ''#useless and used as ages_gender.txt path
index_root = ''
padding = 35
learningrate = 1e-5
momentum = 0.95
weightdecay = 0.05
save_prefix = ""
[validation]
validate = true
data_root = ''
index_root = ''
padding = 35
saveaccuracy = false
saves='re/xct2.npy'
[test]
test = false
data_root = ''
index_root = ''
padding = 35
saveaccuracy = false