最近在调试pytorch代码的时候遇到如下问题,由于他报错的地方不是在我们自己写的代码,而是在pytorch的包里,所以一开始就一头雾水。
在查阅了资料以后http://www.zzvips.com/article/204910.html
里面说报错的地方是位于 __getitem__这个函数当中的,也就是说这里的返回值的维度是不一样的,所以可以在这个函数里面输出返回的值查看即可
比如我这里发现就有明显的不同,然后再对症下药即可。

最近在调试pytorch代码的时候遇到如下问题,由于他报错的地方不是在我们自己写的代码,而是在pytorch的包里,所以一开始就一头雾水。
在查阅了资料以后http://www.zzvips.com/article/204910.html
里面说报错的地方是位于 __getitem__这个函数当中的,也就是说这里的返回值的维度是不一样的,所以可以在这个函数里面输出返回的值查看即可
比如我这里发现就有明显的不同,然后再对症下药即可。