Add changes to the config file

frosinastojanovska · Aug 27, 2024 · b961403 · b961403
1 parent 8b182a5
commit b961403
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/cryosiam/apps/dense_simsiam_regression/config_test.yaml b/cryosiam/apps/dense_simsiam_regression/config_test.yaml
@@ -1,5 +1,4 @@
 data_folder: '/scratch/stojanov/mycoplasma_data/vpp/240422/'
-gt_folder: '/scratch/stojanov/mycoplasma_data/vpp/240422/'
 log_dir: '/scratch/stojanov/development/cryoet-torch/experiments/dense_simsiam_semantic'
 prediction_folder: '/scratch/stojanov/development/cryoet-torch/experiments/dense_simsiam_semantic/predictions'
 trained_model: '/scratch/stojanov/development/cryoet-torch/experiments/dense_simsiam_semantic/model/model-best.ckpt'
@@ -16,13 +15,13 @@ parameters:
   data:
     patch_size: [ 64, 64, 64 ]
     min: 0
-    max: 15
-    mean: 0.5789
-    std: 0.12345
+    max: 1
+    mean: 0
+    std: 1
   network:
     in_channels: 1
     spatial_dims: 3
     n_output_channels: 1
 
 hyper_parameters:
-  batch_size: 40
+  batch_size: 5