
pytorch
Daniel@1994
这个作者很懒,什么都没留下…
展开
-
pycharm ValueError: source code string cannot contain null bytes
pycharm ValueError: source code string cannot contain null bytes 在GitHub下载的代码在win10上调试遇到错误 首先检测python代码中的from,调整目录。如下:用lib.xvision等依次导入需要的函数。 from lib.datasets import GeneralDataset as Dataset from lib.xvision import transforms from lib.xvision.visualizati原创 2021-01-12 15:25:14 · 1763 阅读 · 0 评论 -
pytorch 单gpu改为多gpu训练和测试
pytorch 单gpu改为多gpu训练和测试 训练中 在plfd_backbone = plfd_backbone.to(device)之前加 import torch.nn as nn plfd_backbone = nn.DataParallel(plfd_backbone, device_ids = [0, 1, 2, 3]) device_ids:设置gpu个数。 测试时,同样要修改 在to(device)后加入 pfld_backbone = nn.DataParallel(pfld_ba原创 2020-12-29 11:39:54 · 2443 阅读 · 1 评论