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.
17 lines
348 B
17 lines
348 B
export CUDA_VISIBLE_DEVICES=4
|
|
|
|
|
|
python -u classification.py \
|
|
--task_name "classification" \
|
|
--dataset_name "Heartbeat" \
|
|
--data "UEA" \
|
|
--root_path "./datasets/Heartbeat" \
|
|
--forward_layers 2 \
|
|
--out_channels 32 \
|
|
--stride 16 \
|
|
--batch_size 8 \
|
|
--learning_rate 0.0001 \
|
|
--num_workers 0 \
|
|
--lradj "type1" \
|
|
--patience 5 \
|
|
|