first commit

scripts/long_term_forecast/ETT_script/WPMixer_ETTm2.sh

@@ -0,0 +1,50 @@
+
+# Set the GPU to use
+export CUDA_VISIBLE_DEVICES=0
+
+# Model name
+model_name=WPMixer
+
+# Datasets and prediction lengths
+dataset=ETTm2
+seq_lens=(512 512 512 512)
+pred_lens=(96 192 336 720)
+learning_rates=(0.00076587 0.000275775 0.000234608 0.001039536)
+batches=(256 256 256 256)
+epochs=(80 80 80 80)
+dropouts=(0.4 0.2 0.4 0.4)
+patch_lens=(48 48 48 48)
+lradjs=(type3 type3 type3 type3)
+d_models=(256 256 256 256)
+patiences=(12 12 12 12)
+
+# Model params below need to be set in WPMixer.py Line 15, instead of this script
+wavelets=(bior3.1 db2 db2 db2)
+levels=(1 1 1 1)
+tfactors=(3 3 3 3)
+dfactors=(8 7 5 8)
+strides=(24 24 24 24)
+
+# Loop over datasets and prediction lengths
+for i in "${!pred_lens[@]}"; do
+	python -u run.py \
+		--is_training 1 \
+		--root_path ./data/ETT/ \
+		--data_path ETTm2.csv \
+		--model_id wpmixer \
+		--model $model_name \
+		--task_name long_term_forecast \
+		--data $dataset \
+		--seq_len ${seq_lens[$i]} \
+		--pred_len ${pred_lens[$i]} \
+		--label_len 0 \
+		--d_model ${d_models[$i]} \
+		--patch_len ${patch_lens[$i]} \
+		--batch_size ${batches[$i]} \
+		--learning_rate ${learning_rates[$i]} \
+		--lradj ${lradjs[$i]} \
+		--dropout ${dropouts[$i]} \
+		--patience ${patiences[$i]} \
+		--train_epochs ${epochs[$i]} \
+		--use_amp
+done