用卷积神经网络优化常规BP神经网络
摘要我是个新手,最近做项目遇到了一些问题,老是在将优秀网络模型替换其他网络的时候会出现模型大小不匹配的问题,先将自己做好的代码发出来,做个记录:我用的minist手写字体数据集,因为我的电脑限制,我用CPU跑的,如果条件允许,可以用cuda常规BP神经网络BP网络结构class BP(nn.Module): def __init__(self): super(BP, self).__init__() self.fc = nn.Sequential(
原创
2021-12-27 13:49:16 ·
469 阅读 ·
0 评论