木槿花篱

嘻嘻,欢迎~

摘要: 这是一篇学习帖 对于视觉数据,PyTorch 创建了一个叫做 totchvision 的包,该包含有支持加载类似Imagenet,CIFAR10,MNIST 等公共数据集的数据加载模块 torchvision.datasets 和支持加载图像数据数据转换模块 torch.utils.data.Dat 阅读全文
posted @ 2020-07-29 16:11 木槿花篱 阅读(685) 评论(0) 推荐(0) 编辑
摘要: 这是一篇学习贴。 1 import torch 2 import torch.nn as nn 3 import torch.nn.functional as F 4 import torch.optim as optim 5 from torchvision import datasets, tr 阅读全文
posted @ 2020-07-29 13:41 木槿花篱 阅读(663) 评论(0) 推荐(0) 编辑