torch
torchvision
mne