search

iis-eth-zurich / eeg-tcnet

Apache License 2.0
72 stars 13 forks source link

Hi, is there any pytorch version of eeg-tcnet? #1

Closed chamwen closed 3 years ago

Thoriri commented 3 years ago

Hi!

Since the work was done with Tensorflow and Keras we don't provide any Pytorch version of EEG-TCNet. But following the network architecture figure, you should be able to reproduce the network topology with some minor modifications to count for differences between TF and Torch.

okbalefthanded commented 3 years ago

@chamwen you can find a PyTorch implementation of TCN part in braindecode: https://github.com/braindecode/braindecode/tree/master/braindecode/models

chamwen commented 3 years ago

Thanks!

@chamwen you can find a PyTorch implementation of TCN part in braindecode: https://github.com/braindecode/braindecode/tree/master/braindecode/models

110lmy commented 1 year ago

你好，该代码中的数据集中只有数据的path，label没有相对应的path？

okbalefthanded commented 1 year ago

@110lmy check this notebook, the data labels are created after reading the files. https://github.com/okbalefthanded/eeg-tcnet/blob/master/eeg_tcnet_colab.ipynb