Safa_Mathlouthi:

to_gpu


Hi can you provide the feedback where you found the VoxResNet architecture that you are using?
Also, try to alter your train.py line 90 to read vrn.to('cuda') or vrn.to(0) if this helps.