Skip to content

Commit 7faa1b8

Browse files
committed
fix(doc):2_change_experiment_settings script contents
1 parent af62ec0 commit 7faa1b8

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

docs/tutorial/2_change_experiment_settings.md

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -22,28 +22,34 @@ else
2222
launch_command="accelerate launch"
2323
fi
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
2927
dataset_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

3135
seq_len=3000
3236
for 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 \

0 commit comments

Comments
 (0)