收起
看了下,应该是assert检查这块错了。
assert H=self.img_size[0] and W=self.img_size[1],\
这行可能错了,assert 语句试图把一个 Proxy 对象转换为一个布尔值,这是无法进行的,因为在符号追踪中 Proxy 对象并没有具体的值。在模型的 forward 函数中去掉 assert 语句,或者将其替换为一个可以在符号追踪中处理的操作。
如果有帮助,请点击一下采纳该答案~谢谢
报告相同问题?