diff --git a/cryosiam/apps/dense_simsiam_regression/config_test.yaml b/cryosiam/apps/dense_simsiam_regression/config_test.yaml index 747e347..7085239 100644 --- 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 \ No newline at end of file + batch_size: 5 \ No newline at end of file