cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
493 ワード
cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
メモリが足りないようです.
dataloaders = {x: torch.utils.data.DataLoader(image_datasets[x], batch_size=opt.batchsize,
shuffle=False, num_workers=16) for x in ['gallery','query']}
解決方法:
dataloaders = {x: torch.utils.data.DataLoader(image_datasets[x], batch_size=opt.batchsize, shuffle=False, num_workers=0) for x in ['gallery','query']}