![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
Dawn嗯
这个作者很懒,什么都没留下…
展开
-
Resnet网络详细结构(针对Cifar10)
Resnet网络详细结构(针对Cifar10) 结构 具体结构(Pytorch) conv1 (conv1): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (bn1): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (relu): ReLU(inplace=True) (maxpo原创 2021-07-29 11:27:37 · 529 阅读 · 0 评论 -
pytorch torchvision transform
pytorch torchvision transform Torchvision.Transforms Transforms包含常用图像转换操作。可以使用Compose将它们链接在一起。 此外,还有torchvision.transforms.functional模块,可以对转换进行细粒度控制。 如果需要构建更复杂的图像转换方式,这将很有用。 class torchvision.transforms.Compose(transforms) Example: transforms.Compose([原创 2020-08-11 23:55:03 · 212 阅读 · 0 评论 -
Pytorch指定GPU的方法总结
Pytorch指定GPU的方法 改变系统变量 改变系统环境变量仅使目标显卡,编辑 .bashrc文件,添加系统变量 export CUDA_VISIBLE_DEVICES=0 #这里是要使用的GPU编号 在程序开头设置 os.environ["CUDA_VISIBLE_DEVICES"] = '0,1,2,3' 在运行程序时指定 # 运行程序时使用命令行,来设置该程序可见的gpu: CUDA_VISIBLE_DEVICES=0,1,2,3 python xxx.py 使用torch.c原创 2020-07-21 15:28:49 · 3399 阅读 · 0 评论