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 Datasetfrom lib.xvision import transformsfrom lib.xvision.visualizati原创 2021-01-12 15:25:14 · 1843 阅读 · 0 评论 -
pytorch 单gpu改为多gpu训练和测试
pytorch 单gpu改为多gpu训练和测试训练中在plfd_backbone = plfd_backbone.to(device)之前加import torch.nn as nnplfd_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 · 2481 阅读 · 1 评论
分享