#!/bin/bash

# xPatch_SparseChannel Classification Training Script for FaceDetection Dataset
export CUDA_VISIBLE_DEVICES=0

model_name=xPatch_SparseChannel

python -u run.py \
  --task_name classification \
  --is_training 1 \
  --root_path ./dataset/FaceDetection/ \
  --model_id FaceDetection \
  --model $model_name \
  --data UEA \
  --e_layers 3 \
  --batch_size 64 \
  --seq_len 62 \
  --enc_in 144 \
  --d_model 128 \
  --d_ff 256 \
  --n_heads 8 \
  --patch_len 16 \
  --stride 8 \
  --moving_avg 25 \
  --dropout 0.1 \
  --des 'xPatch_SparseChannel_FaceDetection' \
  --itr 1 \
  --learning_rate 0.0005 \
  --train_epochs 100 \
  --patience 5 \
  --revin 1 \
  --k_graph 8