You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
1.3 KiB
63 lines
1.3 KiB
export CUDA_VISIBLE_DEVICES=0
|
|
|
|
forward_layers=2
|
|
stride=1
|
|
batch_size=4
|
|
lr=0.0001
|
|
|
|
|
|
python -u imputation.py \
|
|
--task_name "imputation" \
|
|
--dataset_name "ETTh2" \
|
|
--data "ETTh2" \
|
|
--root_path "./datasets/ETT/" \
|
|
--data_path "ETTh2.csv" \
|
|
--mask_rate 0.125 \
|
|
--forward_layers $forward_layers \
|
|
--stride 8 \
|
|
--batch_size $batch_size \
|
|
--learning_rate $lr \
|
|
--lradj "cosine" \
|
|
|
|
|
|
python -u imputation.py \
|
|
--task_name "imputation" \
|
|
--dataset_name "ETTh2" \
|
|
--data "ETTh2" \
|
|
--root_path "./datasets/ETT/" \
|
|
--data_path "ETTh2.csv" \
|
|
--mask_rate 0.25 \
|
|
--forward_layers $forward_layers \
|
|
--stride $stride \
|
|
--batch_size 64 \
|
|
--learning_rate $lr \
|
|
--lradj "cosine" \
|
|
|
|
|
|
python -u imputation.py \
|
|
--task_name "imputation" \
|
|
--dataset_name "ETTh2" \
|
|
--data "ETTh2" \
|
|
--root_path "./datasets/ETT/" \
|
|
--data_path "ETTh2.csv" \
|
|
--mask_rate 0.375 \
|
|
--forward_layers $forward_layers \
|
|
--stride $stride \
|
|
--batch_size $batch_size \
|
|
--learning_rate $lr \
|
|
--lradj "type1" \
|
|
|
|
|
|
python -u imputation.py \
|
|
--task_name "imputation" \
|
|
--dataset_name "ETTh2" \
|
|
--data "ETTh2" \
|
|
--root_path "./datasets/ETT/" \
|
|
--data_path "ETTh2.csv" \
|
|
--mask_rate 0.50 \
|
|
--forward_layers 3 \
|
|
--stride $stride \
|
|
--batch_size $batch_size \
|
|
--learning_rate $lr \
|
|
--lradj "type1" \
|