打开torch/nn/functional.py文件,找到torch.batch_norm这一行,将 torch.backends.cudnn.enabled 选项更改为False。 克隆项目.
確定! 回上一頁