77 lines
865 B
YAML
77 lines
865 B
YAML
# environment config
|
|
GPU: "0"
|
|
|
|
|
|
# dataset config
|
|
Path:
|
|
dataset: /home/marques/code/marques/apnea/dataset/BCG_100hz_lowpass50/
|
|
label: ./dataset/
|
|
save: ./output/
|
|
|
|
batch_size: 256
|
|
number_worker: 0
|
|
model_name: HYBRIDNET018
|
|
select_sampno:
|
|
- 88
|
|
- 220
|
|
- 221
|
|
- 229
|
|
- 282
|
|
- 286
|
|
- 541
|
|
- 579
|
|
- 582
|
|
- 670
|
|
- 671
|
|
- 683
|
|
- 703
|
|
- 704
|
|
- 726
|
|
- 735
|
|
- 736
|
|
- 933
|
|
- 935
|
|
- 952
|
|
- 954
|
|
- 955
|
|
- 960
|
|
- 961
|
|
- 962
|
|
- 966
|
|
- 967
|
|
- 969
|
|
- 971
|
|
- 972
|
|
- 1000
|
|
- 1004
|
|
- 1006
|
|
- 1009
|
|
- 1010
|
|
- 1296
|
|
- 1300
|
|
- 1301
|
|
- 1302
|
|
- 1308
|
|
- 1354
|
|
- 1374
|
|
- 1378
|
|
- 1478
|
|
|
|
|
|
# train hyperparameters config
|
|
epoch: 100
|
|
lr: 0.0001
|
|
nc: 1
|
|
|
|
# wandb config
|
|
entity: "marques"
|
|
project: "Sleep_Apnea_HYBRID00X"
|
|
Note: "HYBRID010 RESP "
|
|
tags: ["ReduceLROnPlateau", "RESP LSTM"]
|
|
|
|
|
|
# "CW":class_weight
|
|
|
|
|
|
# "CosineAnnealingLR"
|
|
# "ReduceLROnPlateau" |