VolUnD-Toolkit/trainingSetup.txt at main · EUROVOLC-ML/VolUnD-Toolkit · GitHub

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
##############################################################################
# TRAINING SETUP: please don't delete or modify attributes name (before ':') #
##############################################################################

# Dataset options
train_dir: './dataset/trainingSet'
val_dir: './dataset/validationSet'
data_location: 'path/to/data/'
chunk_len: 512
chunk_only_one: False
chunk_rate: 1
chunk_random_crop: False
data_sampling_frequency: None
chunk_linear_subsample: 1
chunk_butterworth_lowpass: None
chunk_butterworth_highpass: None
chunk_butterworth_order: 2
channels_list: None
channels_name: None
batch_size: 128
data_provider: 'ram'
mean: None
std: None
training_labels: [0]
validation_labels: None

# Experiment options
tag: 'ae'
log_dir: './logs'
plot_every: 100
log_every: 10
save_every: 10
tensorboard_enable: True
tensorboard_port: 6006

# Model options
layers_base: 1
channels_base: 16
min_spatial_size: 2
start_dilation: 3
min_sig_dil_ratio: 50
max_channels: 1024
h_size: 64
enable_variational: False

# Training options
optim: 'Adam'
reduce_lr_every: None
reduce_lr_factor: 0.1
weight_decay: 0.0005
epochs: 32000
lr: 0.00001
device: 'cuda'

# Checkpoints restore options
checkpoint: None