随笔分类 -  Pytorch

Pytorch使用笔记
摘要:Resnet网络详细结构(针对Cifar10) 结构 具体结构(Pytorch) conv1 (conv1): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (bn1): BatchNorm2 阅读全文
posted @ 2021-07-28 23:30 Dawn嗯 阅读(273) 评论(0) 推荐(0)
摘要:pointnet.pytorch代码解析 阅读全文
posted @ 2021-01-04 17:47 Dawn嗯 阅读(700) 评论(0) 推荐(0)
摘要:Torchvision.Transforms Transforms包含常用图像转换操作。可以使用Compose将它们链接在一起。 此外,还有torchvision.transforms.functional模块,可以对转换进行细粒度控制。 如果需要构建更复杂的图像转换方式,这将很有用。 class 阅读全文
posted @ 2020-08-11 23:46 Dawn嗯 阅读(226) 评论(0) 推荐(0)
摘要:Pytorch指定GPU的方法 改变系统变量 改变系统环境变量仅使目标显卡,编辑 .bashrc文件,添加系统变量 export CUDA_VISIBLE_DEVICES=0 #这里是要使用的GPU编号 在程序开头设置 os.environ["CUDA_VISIBLE_DEVICES"] = '0, 阅读全文
posted @ 2020-07-21 15:29 Dawn嗯 阅读(1843) 评论(0) 推荐(0)