search

pquochuy / xsleepnet

50 stars 9 forks source link

Regarding hidden state size of GRU #6

Open andyjung12 opened 11 months ago

andyjung12 commented 11 months ago

In the paper, it said that the hidden state size of GRU for inter-epoch level of raw data is 64. But in the config that you used in the github here says that it is 256. Which one is correct?

pquochuy commented 11 months ago

There are two of of them, 1 for seqsleepnet branch and 1 for cnn-rnn brach. The former is one is 64, the latter is 256.

andyjung12 commented 11 months ago

so, in the paper it said, "The LSTM and GRU cells responsible for inter-epoch sequential modelling had 256 and 64 units in their hidden state vectors, respectively"

It is other way around, right? 64 for seqsleepnet and 256 for cnn-rnn branch?

pquochuy commented 11 months ago

Yes, thats right.