[304dd3]: / config.yaml

Download this file

46 lines (43 with data), 819 Bytes

 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
file_name: ./saved_model/1007_pooling_age
data:
data_root_dir: data/
dir_to_scans: dir_to_scans
dir_to_tsv: ./datasets/files
batch_size: 4
val_batch_size: 2
workers: 4
percentage_usage: 1.0
exp_name: 1007_pooling_age
visdom:
port: 8064
server: skygpu07
model:
arch: ours
input_channel: 1
nhid: 512
feature_dim: 1024
n_label: 3
expansion: 8
num_blocks: 0
type_name: conv3x3x3
norm_type: Instance
adv_model:
nhid: 36
out_dim: 12
mmse_model:
nhid: 64
training_parameters:
use_age: False
pretrain: #dir to saved model here
max_iter: 16000
start_epoch: 0
epochs: 200
print_freq: 10
max_grad_l2_norm:
report_interval: 100
snapshot_interval: 1000
optimizer:
method: SGD
par:
lr: 0.01
weight_decay: 0.000