pytorch
andrew80
这个作者很懒,什么都没留下…
展开
-
pytorch中多GPU训练
如果指定在某个GPU(如0号GPU)上训练,则只需要添加:import osos.environ["CUDA_VISIBLE_DEVICES"] = "0"如果需要同时使用多个GPU,则参考:https://oldpan.me/archives/pytorch-to-use-multiple-gpusPytorch中多GPU训练指北前言在数据越来越多的时代,随着模型规...转载 2019-04-10 17:51:58 · 16586 阅读 · 1 评论 -
显示数据集图片并计算数据集的transforms.Normalize
import torchimport torchvisionfrom torch.utils.data import DataLoaderfrom torchvision import transforms, datasetsimport matplotlib.pyplot as pltimport numpy as npdata_transform = transforms.Co...原创 2019-06-12 23:07:19 · 2260 阅读 · 0 评论