20 lines
385 B
Bash
20 lines
385 B
Bash
|
export CUDA_VISIBLE_DEVICES=6
|
||
|
|
|
||
|
|
python -u run.py \
|
||
|
|
--task_name anomaly_detection \
|
||
|
|
--is_training 1 \
|
||
|
|
--root_path ./dataset/PSM \
|
||
|
|
--model_id PSM \
|
||
|
|
--model Autoformer \
|
||
|
|
--data PSM \
|
||
|
|
--features M \
|
||
|
|
--seq_len 100 \
|
||
|
|
--pred_len 0 \
|
||
|
|
--d_model 128 \
|
||
|
|
--d_ff 128 \
|
||
|
|
--e_layers 3 \
|
||
|
|
--enc_in 25 \
|
||
|
|
--c_out 25 \
|
||
|
|
--anomaly_ratio 1 \
|
||
|
|
--batch_size 128 \
|
||
|
|
--train_epochs 3
|