Diff of /config/config.yaml [000000] .. [3ed61c]

Switch to unified view

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