catanddog_自定义CNN分类与测试
发布时间
阅读量：
阅读量
1、采用的自定义卷积神经网络
设计参考了
1.1神经网络结构
输入图像的尺寸设定为112乘112，整体架构采用了包含四层卷积层与两层输出层的设计，各层卷积模块的具体构造如下
1.代码实现与功能验证
(1)在使用CrossEntropyLoss()作为损失函数时，输出层无需进行额外设置
criteon = nn.CrossEntropyLoss()
......................
.....................
loss = criteon(logits, y)
(2)全部源代码
# import全部评论 (0)
还没有任何评论哟~