Did you get problem running with GPU?

[longnguyen2]

I tried to run your code but it stucked at line 13 of Trainer.py self.network = self.network.to(device), it said RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at ..\aten\src\THC\THCGeneral.cpp:50

Any special config? I'm currently using pytorch 1.4.0

[Adesoji1]

upgrade your pytorch version and set your device to use gpu