search

vlawhern / arl-eegmodels

This is the Army Research Laboratory (ARL) EEGModels Project: A Collection of Convolutional Neural Network (CNN) models for EEG signal classification, using Keras and Tensorflow
Other
1.17k stars 287 forks source link

The implement of PyTorch #24

Closed BIRlz closed 3 years ago

BIRlz commented 3 years ago

I am wondering whether you would release a pytorch implement of this repo? If possibile, it will be a great convenient for us!

vlawhern commented 3 years ago

All of the models implemented here are also implemented in the braindecode package (which uses PyTorch as the backend), available at https://github.com/braindecode/braindecode

BIRlz commented 3 years ago

All of the models implemented here are also implemented in the braindecode package (which uses PyTorch as the backend), available at https://github.com/braindecode/braindecode

Thinks for your work and reply! Have a nice day!