记录一下:
torch.backends.cudnn.enabled = False
this solution might increase the memory usage since it will disable cudnn for other modules like Cov2d.
引用自:https://github.com/traveller59/second.pytorch/issues/293
记录一下:
torch.backends.cudnn.enabled = False
this solution might increase the memory usage since it will disable cudnn for other modules like Cov2d.
引用自:https://github.com/traveller59/second.pytorch/issues/293