|
a |
|
b/config/config.yaml |
|
|
1 |
Network: |
|
|
2 |
seed: 1 |
|
|
3 |
batch_size: 1 # number of 3D images to be loaded in the data loader |
|
|
4 |
valid_batch_size: 1 |
|
|
5 |
test_batch_size: 1 # must be dividable by 30 (1, 2, 3, 4, 5, 6, 10, 15, 30) |
|
|
6 |
lr: 1e-4 |
|
|
7 |
weight_decay: 0 |
|
|
8 |
amsgrad: True |
|
|
9 |
resize_shape_h_2D3D: 150 |
|
|
10 |
resize_shape_w_2D3D: 110 |
|
|
11 |
network_output_path: network_data/ |
|
|
12 |
output_data_path: output_data/ |
|
|
13 |
tb_logs_path: tensor_board_logs/ |
|
|
14 |
stat_log_path: stat_logs |
|
|
15 |
checkpoint_name: checkpoint.tar |
|
|
16 |
trained_model_name: trained_model.pth |
|
|
17 |
|
|
|
18 |
# changeable items: |
|
|
19 |
target_dir: /home/soroosh/Documents/Repositories_target_files/ACL_tear/ |
|
|
20 |
network_save_freq: 500 # the real save frequency will be this number times 3 |
|
|
21 |
network_checkpoint_freq: 2 |
|
|
22 |
valid_iteration_freq: 75 |
|
|
23 |
num_iterations: 1000 # number of epochs |