Remove unnecessary arguments --in-channels and --out-channels
This commit is contained in:
parent
f64b1e42e9
commit
0764a1006e
8 changed files with 17 additions and 10 deletions
|
@ -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" \
|
||||
--in-channels 3 --out-channels 3 --norms cosmology.dis \
|
||||
--norms cosmology.dis \
|
||||
--batches 1 --loader-workers 0 --pad-or-crop 40 \
|
||||
--load-state best_model.pth
|
||||
|
||||
|
|
|
@ -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" \
|
||||
--in-channels 3 --out-channels 3 --norms cosmology.dis --augment \
|
||||
--norms cosmology.dis --augment \
|
||||
--epochs 1024 --batches 3 --loader-workers 3 --lr 0.001
|
||||
# --load-state checkpoint.pth
|
||||
|
||||
|
|
|
@ -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" \
|
||||
--in-channels 3 --out-channels 3 --norms cosmology.vel \
|
||||
--norms cosmology.vel \
|
||||
--batches 1 --loader-workers 0 --pad-or-crop 40 \
|
||||
--load-state best_model.pth
|
||||
|
||||
|
|
|
@ -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" \
|
||||
--in-channels 3 --out-channels 3 --norms cosmology.vel --augment \
|
||||
--norms cosmology.vel --augment \
|
||||
--epochs 1024 --batches 3 --loader-workers 3 --lr 0.001
|
||||
# --load-state checkpoint.pth
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue