2022年1月31日
torch.nn.modules类构建模型
摘要： 一、torch.nn.modules 网络模型构建于一系列：nn.Conv2d、AvgPool2d、ReLU6、BatchNorm2d、CrossEntropyLoss等，由此基本操作组成。先看操作类型，再看操作组成方式。 1、属于nn.modules的操作类型 ① class torch.nn.C 阅读全文
posted @ 2022-01-31 00:14 shines87 阅读(428) 评论(0) 推荐(0) 编辑