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

浙公网安备 33010602011771号