Move UNet with ResBlock to VNet and Revert UNet to the previous simple version

scripts/dis2dis-test.slurm

@@ -37,7 +37,7 @@ tgt_files="$files"
 m2m.py test \
     --test-in-patterns "$data_root_dir/$in_dir/$test_dirs/$in_files" \
     --test-tgt-patterns "$data_root_dir/$tgt_dir/$test_dirs/$tgt_files" \
-    --norms cosmology.dis \
+    --norms cosmology.dis --model VNet \
     --batches 1 --loader-workers 0 --pad-or-crop 40 \
     --load-state best_model.pth
 

scripts/dis2dis.slurm

@@ -42,7 +42,7 @@ srun m2m.py train \
     --train-tgt-patterns "$data_root_dir/$tgt_dir/$train_dirs/$tgt_files" \
     --val-in-patterns "$data_root_dir/$in_dir/$val_dirs/$in_files" \
     --val-tgt-patterns "$data_root_dir/$tgt_dir/$val_dirs/$tgt_files" \
-    --norms cosmology.dis --augment \
+    --norms cosmology.dis --augment --model VNet \
     --epochs 1024 --batches 3 --loader-workers 3 --lr 0.001
 #    --load-state checkpoint.pth
 

scripts/vel2vel-test.slurm

@@ -37,7 +37,7 @@ tgt_files="$files"
 m2m.py test \
     --test-in-patterns "$data_root_dir/$in_dir/$test_dirs/$in_files" \
     --test-tgt-patterns "$data_root_dir/$tgt_dir/$test_dirs/$tgt_files" \
-    --norms cosmology.vel \
+    --norms cosmology.vel --model VNet \
     --batches 1 --loader-workers 0 --pad-or-crop 40 \
     --load-state best_model.pth
 

scripts/vel2vel.slurm

@@ -42,7 +42,7 @@ srun m2m.py train \
     --train-tgt-patterns "$data_root_dir/$tgt_dir/$train_dirs/$tgt_files" \
     --val-in-patterns "$data_root_dir/$in_dir/$val_dirs/$in_files" \
     --val-tgt-patterns "$data_root_dir/$tgt_dir/$val_dirs/$tgt_files" \
-    --norms cosmology.vel --augment \
+    --norms cosmology.vel --augment --model VNet \
     --epochs 1024 --batches 3 --loader-workers 3 --lr 0.001
 #    --load-state checkpoint.pth