-
Notifications
You must be signed in to change notification settings - Fork 17
/
Copy pathoptions_ind.toml
57 lines (51 loc) · 987 Bytes
/
options_ind.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
title = "covid19 PyTorch"
[general]
usecudnn = true
usecudnnbenchmark = true
gpuid = "7"
loadpretrainedmodel = true
random_seed = true
pretrainedmodelpath = 'weights/ind_model.pt'
use_3d = false
use_lstm=false
use_slice=true
use_plus=false
use25d=false
plus_as_input=false
mod='ind'#'ab
class_num=3
[input]
batchsize = 32
numworkers = 8
shuffle = true
[model]
type = "Finetune-label"
inputdim = 256
hiddendim = 256
numclasses = 2
numlstms = 2
[training]
train = false
epochs = 31
startepoch = 0
statsfrequency = 1000
data_root = ''#useless and used as ages_gender.txt path
index_root = 'data/txt/ind_train_jpg.txt'
padding = 35
learningrate = 1e-5
momentum = 0.95
weightdecay = 0.05
save_prefix = "weights/ind_model"
[validation]
validate = true
data_root = ''
index_root = 'ind'
padding = 35
saveaccuracy = false
saves='re/ind.npy'
[test]
test = false
data_root = '/mnt/data6/lung_jpgs_with_SEG'
index_root = 'data/txt/2train_lidc_ild_train.txt'
padding = 45
saveaccuracy = false