morgoth/train_EEG_level_head.sh at main · bdsp-core/morgoth · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
#!/bin/bash

num_epochs=10

echo "exxact@1" | sudo -S  $(which python) EEG_level_head.py \
        --mode train \
        --dataset NORMAL \
        --train_csv_dirs "/data/NORMAL/EEG_level/continuous_results /data/MoE/results/pred_NORMAL_100pStep /data/Sandor_100/results_polarity/pred_NORMAL_1sStep" \
        --file_list_path /data/NORMAL/EEG_level/training_list.csv \
        --output_dir checkpoints/EEG_level_NORMAL6  \
        --num_epochs ${num_epochs} \
        --pe_max_length 15000 \
        --lr 5e-4 \
        --save_freq 1 \
        --resume_training

# --train_csv_dirs "/data/FOC_GEN_SPIKES/EEG_level/continuous_results /data/SPIKES/FOC_GEN_SPIKES/continuous_results_control /data/MoE/results/pred_FOCGENSPIKES_100pStep /data/Sandor_100/results/pred_FOCGENSPIKES_1sStep /data/OccasionNoise/results/pred_FOCGENSPIKES_1sStep" \

SLOWING_datasets=("FOC_SLOWING" "GEN_SLOWING")
for SLOWING_dataset in "${SLOWING_datasets[@]}"; do
  echo "exxact@1" | sudo -S  $(which python) EEG_level_head.py \
          --mode train \
          --dataset ${SLOWING_dataset} \
          --train_csv_dirs "/data/SLOWING/EEG_level/continuous_results /data/MoE/results/pred_SLOWING_100pStep /data/Sandor_100/results_polarity/pred_SLOWING_1sStep /data/OccasionNoise/results/pred_SLOWING_1sStep" \
          --file_list_path /data/SLOWING/EEG_level/training_list.csv \
          --output_dir checkpoints/EEG_level_${SLOWING_dataset}6  \
          --num_epochs ${num_epochs} \
          --pe_max_length 15000 \
          --focal_alpha "0.25" \
          --lr 5e-4 \
          --save_freq 1 \
          --resume_training
done


echo "exxact@1" | sudo -S  $(which python) EEG_level_head.py \
        --mode train \
        --dataset BS \
        --train_csv_dirs "/data/BS/EEG_level/continuous_results /data/MoE/results/pred_BS_100pStep" \
        --file_list_path /data/BS/EEG_level/training_list.csv \
        --output_dir checkpoints/EEG_level_BS6  \
        --num_epochs ${num_epochs} \
        --pe_max_length 15000 \
        --focal_alpha "0.25" \
        --lr 5e-4 \
        --save_freq 1 \
        --resume_training

#  --train_csv_dirs "/data/BS/EEG_level/continuous_results /data/BS/EEG_level/continuous_results_control /data/MoE/results/pred_BS_100pStep" \

echo "exxact@1" | sudo -S  $(which python) EEG_level_head.py \
        --mode train \
        --dataset SPIKES \
        --train_csv_dirs "/data/SPIKES/EEG_level/continuous_results /data/HEP/HEP_events/pred_1sStep  /data/HEP/HEP_1h_Data/pred_1sStep /data/Sandor_100/results_polarity/pred_SPIKES_1sStep /data/OccasionNoise/results/pred_SPIKES_1sStep" \
        --file_list_path /data/SPIKES/EEG_level/training_list.csv \
        --output_dir checkpoints/EEG_level_SPIKES6  \
        --num_epochs ${num_epochs} \
        --pe_max_length 15000 \
        --lr 5e-4 \
        --save_freq 1 \
        --resume_training

#  --train_csv_dirs "/data/SPIKES/EEG_level/continuous_results /data/SPIKES/EEG_level/continuous_results_control /data/HEP/HEP_events/pred_1sStep  /data/HEP/HEP_1h_Data/pred_1sStep /data/Sandor_100/results/pred_SPIKES_1sStep /data/OccasionNoise/results/pred_SPIKES_1sStep" \

FOC_GEN_SPIKES_datasets=("FOC_SPIKES" "GEN_SPIKES")
for FOC_GEN_SPIKES_dataset in "${FOC_GEN_SPIKES_datasets[@]}"; do
  echo "exxact@1" | sudo -S  $(which python) EEG_level_head.py \
          --mode train \
          --dataset ${FOC_GEN_SPIKES_dataset} \
          --train_csv_dirs "/data/FOC_GEN_SPIKES/EEG_level/continuous_results /data/MoE/results/pred_FOCGENSPIKES_100pStep /data/Sandor_100/results_polarity/pred_FOCGENSPIKES_1sStep /data/OccasionNoise/results/pred_FOCGENSPIKES_1sStep" \
          --file_list_path /data/FOC_GEN_SPIKES/EEG_level/training_list.csv \
          --output_dir checkpoints/EEG_level_${FOC_GEN_SPIKES_dataset}6 \
          --num_epochs ${num_epochs} \
          --pe_max_length 15000 \
          --focal_alpha "0.25" \
          --lr 5e-4 \
          --save_freq 1 \
          --resume_training
done


# --train_csv_dirs "/data/SLOWING/EEG_level/continuous_results /data/SLOWING/EEG_level/continuous_results_control /data/MoE/results/pred_SLOWING_100pStep /data/Sandor_100/results/pred_SLOWING_1sStep /data/OccasionNoise/results/pred_SLOWING_1sStep" \

#IIIC_datasets=("SEIZURE" "GPD" "LPD" "LRDA" "GRDA") #
#for IIIC_dataset in "${IIIC_datasets[@]}"; do
#  echo "exxact@1" | sudo -S  $(which python) EEG_level_head.py \
#          --mode train \
#          --dataset ${IIIC_dataset} \
#          --train_csv_dirs "/data/IIIC/EEG_level/continuous_results /data/MoE/results/pred_IIIC_100pStep" \
#          --file_list_path /data/IIIC/EEG_level/training_list.csv \
#          --output_dir checkpoints/EEG_level_${IIIC_dataset}6  \
#          --num_epochs ${num_epochs} \
#          --pe_max_length 15000 \
#          --focal_alpha "0.25" \
#          --lr 5e-4 \
#          --save_freq 1 \
#          --resume_training \
#          --batch_size 100
#done

# --train_csv_dirs "/data/IIIC/EEG_level/continuous_results /data/IIIC/EEG_level/continuous_results_control /data/MoE/results/pred_IIIC_100pStep" \