Pytorch AssertionError问题解决

今天跑DaNN源代码时遇到了一个报错：
出错代码：

assert(X.size(0)==Y.size(0))

啥也不懂的萌新先查了一下assert()是个啥：assert宏的原型定义在<assert.h>中，其作用是如果它的条件返回错误，则终止程序执行。assert()常被用在函数开始处检验传入参数的合法性。

参考博客https://blog.csdn.net/qq_29931083/article/details/102582018，报错可能有两个问题：①数据类型不是tensor ②输入维度不一致
先在报错行前把X，Y类型和维度分别打印出来检查一下：


果然在某一次迭代时，X，Y维度不一致了。原来是数据集不能正好被batch（64）整除，所以最后一块数据集只有39个图片。为了先把代码跑通，先把39个数据舍去，提前一步让batch_j置零，读取不到39个图片的那一块：

跑通