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

Switch to side-by-side view

--- a
+++ b/config/config.yaml
@@ -0,0 +1,23 @@
+Network:
+  seed: 1
+  batch_size: 1 # number of 3D images to be loaded in the data loader
+  valid_batch_size: 1
+  test_batch_size: 1 # must be dividable by 30 (1, 2, 3, 4, 5, 6, 10, 15, 30)
+  lr: 1e-4
+  weight_decay: 0
+  amsgrad: True
+  resize_shape_h_2D3D: 150
+  resize_shape_w_2D3D: 110
+network_output_path: network_data/
+output_data_path: output_data/
+tb_logs_path: tensor_board_logs/
+stat_log_path: stat_logs
+checkpoint_name: checkpoint.tar
+trained_model_name: trained_model.pth
+
+# changeable items:
+target_dir: /home/soroosh/Documents/Repositories_target_files/ACL_tear/
+network_save_freq: 500 # the real save frequency will be this number times 3
+network_checkpoint_freq: 2
+valid_iteration_freq: 75
+num_iterations: 1000 # number of epochs
\ No newline at end of file