2021
04-25
04-25
pytorch tensor int型除法出现的问题
昨天晚上跑起来一个classification实验,今天发现训练loss在降,然而accuracy永远是0。。。直觉告诉我evaluation有问题然后发现自己写了个很愚蠢的bugaccuracy对应的tensor出来是int型的,我用到了一个除法取平均。而pytorch里无论用/or//结果都是取整,,accuracy一直就没有。。所以转换成float就没问题了,,低级bug,下不为例补充:pytorchtensordivision/除法除法的时候要注意数据类型的问题a=torch.div(torch.tensor...
继续阅读 >