Extend training epochs in example scripts

scripts/dis2den.slurm

@@ -42,7 +42,7 @@ srun m2m.py train \
     --in-norms cosmology.dis --tgt-norms torch.log1p --augment --crop 128 --pad 20 \
     --model UNet \
     --lr 0.0001 --batches 1 --loader-workers 0 \
-    --epochs 128 --seed $RANDOM \
+    --epochs 1024 --seed $RANDOM \
     --cache --div-data
 #    --load-state checkpoint.pth \
 

scripts/dis2dis.slurm

@@ -43,7 +43,7 @@ srun m2m.py train \
     --in-norms cosmology.dis --tgt-norms cosmology.dis --augment --crop 128 --pad 20 \
     --model VNet --adv-model UNet --cgan \
     --lr 0.0001 --adv-lr 0.0004 --batches 1 --loader-workers 0 \
-    --epochs 128 --seed $RANDOM \
+    --epochs 1024 --seed $RANDOM \
     --cache --div-data
 #    --load-state checkpoint.pth \
 

scripts/srsgan.slurm

@@ -43,7 +43,7 @@ srun m2m.py train \
     --in-norms cosmology.dis,cosmology.vel --tgt-norms cosmology.dis,cosmology.vel --augment --crop 88 --pad 20 --scale-factor 2 \
     --model VNet --adv-model PatchGAN --cgan \
     --lr 0.0001 --adv-lr 0.0004 --batches 1 --loader-workers 0 \
-    --epochs 128 --seed $RANDOM \
+    --epochs 1024 --seed $RANDOM \
     --cache --div-data
 #    --val-in-patterns "$data_root_dir/$in_dir/$val_dirs/$in_files_1,$data_root_dir/$in_dir/$val_dirs/$in_files_2" \
 #    --val-tgt-patterns "$data_root_dir/$tgt_dir/$val_dirs/$tgt_files_1,$data_root_dir/$tgt_dir/$val_dirs/$tgt_files_2" \

scripts/vel2vel.slurm

@@ -43,7 +43,7 @@ srun m2m.py train \
     --in-norms cosmology.vel --tgt-norms cosmology.vel --augment --crop 128 --pad 20 \
     --model VNet --adv-model UNet --cgan \
     --lr 0.0001 --adv-lr 0.0004 --batches 1 --loader-workers 0 \
-    --epochs 128 --seed $RANDOM \
+    --epochs 1024 --seed $RANDOM \
     --cache --div-data
 #    --load-state checkpoint.pth \