pytorch tensor int型除法

昨天晚上跑起来一个classification实验,今天发现训练loss在降,然而accuracy永远是0 。。。直觉告诉我evaluation有问题 然后发现自己写了个很愚蠢的bug accuracy对应的tensor出来是int型的,我用到了一个除法取平均。而pytorch里无论用 / or // 结果都是取整,,accuracy一直就没有。。 所以转换成float就没问题了,,低级bug,下
相关文章
相关标签/搜索