pquochuy / MultitaskSleepNet

Joint Classification and Prediction CNN Framework for Automatic Sleep Stage Classification
39 stars 19 forks source link

How to train the CNN filters #1

Closed shqing closed 6 years ago

shqing commented 6 years ago

Hello, if i want to train the filterbanks of CNN, how should I do? By executing the script tensorflow_net/dnn-filterbank/train_dnn_filterbank_gpu0 ? How this script is to train the filters of DNN.

pquochuy commented 6 years ago

Hi shqing,

To train the DNN filterbanks, you have first download the datasets and prepare the data as I described in Step 1 and 2. Afterwards, you can execute the following script:

If you view the script file, you can see they basically call train_dnn_filterbank_gpu0.py and test_dnn_filterbank_gpu0.py

Hope this would help and please let me know if something goes wrong.

Huy