我有一个CNN 通过TensorFlow实现图像分类，现在用 PyTorch 复现遇到问题怎么办？

笨熊猫  

2017-12-05

有问题想请教一下，现在我有一个CNN实现的图像分类，用的框架是tensorflow，但是现在我想用Pytorch去复现，在读取图片的时候报错RuntimeError: inconsistent tensor sizes at，我觉得是在读取的过程中DataLoader里面参数的问题，但是不知道如何解决，不知道有大神能够帮我一下吗？谢谢各位！

 

我是用了transfroms，我是依照MNIST_CNN 的读入数据弄得。

 

 

这是数据形式 里面的图片是不规则像素大小的

 

 

 收藏  分享
机器学习 深度学习 PyTorch Tensorflow 神经网络
关注问题 3人已关注
 写回答 2人已回答
2个回答

暂时没有回答，开始写第一个回答