Wednesday 14 August 2019

RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:405

disable the following line,
# cudnn.benchmark = True

No comments:

Post a Comment