42 lines
607 B
YAML
42 lines
607 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: HYBRIDNET003
|
|
select_sampno:
|
|
- 282
|
|
- 726
|
|
- 229
|
|
- 582
|
|
- 286
|
|
- 966
|
|
- 1000
|
|
- 1004
|
|
- 1006
|
|
- 1009
|
|
|
|
# train hyperparameters config
|
|
epoch: 50
|
|
lr: 0.00001
|
|
nc: 1
|
|
|
|
# wandb config
|
|
entity: "marques"
|
|
project: "Sleep_Apnea_HYBRID00X"
|
|
Note: "HYBRID003 BD STFT RESP DUAL"
|
|
tags: ["ReduceLROnPlateau", "RESP CNN", "STFT LSTM"]
|
|
|
|
|
|
# "CW":class_weight
|
|
|
|
|
|
# "CosineAnnealingLR"
|
|
# "ReduceLROnPlateau" |