摘要: 文章目录 1.获取训练设备2.定义类3.模型层4.模型参数 神经网络由对数据执行操作的层/模块组成。 torch.nn 命名空间提供了构建自己的神经网络所需的所有构建块。PyTorch 中的每个模块都是 nn.Module 的子类。神经网络是一个模块本身,由其他模块(层)组成。这种嵌套结构允许轻松构 阅读全文
posted @ 2021-08-17 17:20 小Aer 阅读(4) 评论(0) 推荐(0) 编辑
摘要: 文章目录 1.Transfroms2.ToTensor()3.Lambda Transforms 1.Transfroms 数据并不总是以训练机器学习算法所需的最终处理形式出现。我们使用转换来对数据进行一些操作并使其适合训练。 所有 TorchVision 数据集都有两个参数 - 用于修改特征的 t 阅读全文
posted @ 2021-08-17 16:32 小Aer 阅读(9) 评论(0) 推荐(0) 编辑