摘要: 首先,必须继承nn.Module这个类,要让Pytorch知道这个类是一个Module。 其次,在_init_(self)中设置需要的组件,比如(Conv,Pooling,Linear,BatchNorm等) 最后,在forward(self,x)中用定义好的组件进行组装,就像搭积木,把网络结构搭建 阅读全文
posted @ 2019-09-18 17:27 珠峰上吹泡泡 阅读(413) 评论(0) 推荐(0) 编辑
摘要: 01 引入 假如我们想要在cifar10上验证vgg的性能,可能就会出现以下问题: vgg 标准输入:224 x 224 x3 cifar10图像的分辨率:32 x 32 x 3 02 可行的解决方法 一种观点: 修改最后几个全连接层的大小!(绝对可行) 另一种观点: 对于普通的图像分类,resiz 阅读全文
posted @ 2019-09-18 17:13 珠峰上吹泡泡 阅读(848) 评论(0) 推荐(0) 编辑