20 lines
394 B
Bash
20 lines
394 B
Bash
export CUDA_VISIBLE_DEVICES=0
|
|
|
|
python -u run.py \
|
|
--task_name anomaly_detection \
|
|
--is_training 1 \
|
|
--root_path ./dataset/SMD \
|
|
--model_id SMD \
|
|
--model KANAD \
|
|
--data SMD \
|
|
--features M \
|
|
--seq_len 96 \
|
|
--d_model 4 \
|
|
--enc_in 38 \
|
|
--c_out 38 \
|
|
--anomaly_ratio 0.5 \
|
|
--learning_rate 0.01 \
|
|
--batch_size 128 \
|
|
--num_workers 4 \
|
|
--patience 5 \
|
|
--train_epochs 100 |