The fusion of MNE and PyTorch for accelerated deep-neural-network based BCI-systems and Neurophysiology signal analysis.
BSD 3-Clause "New" or "Revised" License
64
stars
19
forks
source link
GPU memory fills during prediction/evaluation #38
Closed
kostasde closed 4 years ago
move outputs/predictions/labels to the cpu after finished with model in base process to spare GPU memory