File tree Expand file tree Collapse file tree 1 file changed +12
-6
lines changed
Expand file tree Collapse file tree 1 file changed +12
-6
lines changed Original file line number Diff line number Diff line change 2222 launch_command=" accelerate launch"
2323fi
2424
25- model_name= " $( basename " $( dirname " $( readlink -f " $0 " ) " ) " ) " # folder name
25+ . " $( dirname " $( readlink -f " $0 " ) " ) /../globals.sh " # Import shared information from scripts/globals.sh
2626
27- dataset_root_path=storage/datasets/HumanActivity
28- model_id=$model_name
2927dataset_name=$( basename " $0 " .sh) # file name
28+ dataset_subset_name=" "
29+ dataset_id=$dataset_name
30+ get_dataset_info " $dataset_name " " $dataset_subset_name " # Get dataset information from scripts/globals.sh
31+
32+ model_name=" $( basename " $( dirname " $( readlink -f " $0 " ) " ) " ) " # folder name
33+ model_id=$model_name
3034
3135seq_len=3000
3236for pred_len in 300; do
3337 $launch_command main.py \
3438 --is_training 1 \
39+ --collate_fn " collate_fn" \
3540 --loss " ModelProvidedLoss" \
3641 --use_multi_gpu $use_multi_gpu \
3742 --dataset_root_path $dataset_root_path \
3843 --model_id $model_id \
3944 --model_name $model_name \
4045 --dataset_name $dataset_name \
46+ --dataset_id $dataset_id \
4147 --features M \
4248 --seq_len $seq_len \
4349 --pred_len $pred_len \
44- --enc_in 12 \
45- --dec_in 12 \
46- --c_out 12 \
50+ --enc_in $n_variables \
51+ --dec_in $n_variables \
52+ --c_out $n_variables \
4753 --train_epochs 300 \
4854 --patience 10 \
4955 --val_interval 1 \
You can’t perform that action at this time.
0 commit comments