DBNet.pytorch icon indicating copy to clipboard operation
DBNet.pytorch copied to clipboard

Published 20 hours ago • verified publisher icon WenmuZhou

Cudnn error in validation step

Open hzhuangdy opened this issue 4 years ago • 1 comments

Hi,

I wonder why this happened after a few iters of validation?

cudnn(1)

hzhuangdy avatar Aug 04 '20 11:08 hzhuangdy

Hi Use torch.backends.cudnn.enabled=False in the first of_eval function in trainer.py.

smama65 avatar May 18 '22 06:05 smama65