1. 首页
  2. 编程语言
  3. Python
  4. pytorch tensor int型除法出现的问题

pytorch tensor int型除法出现的问题

上传者: 2022-02-08 09:27:09上传 PDF文件 66.50 KB 热度 15次

昨天晚上跑起来一个classification实验,今天发现训练loss在降,然而accuracy永远是0 。。。直觉告诉我evaluation有问题然后发现自己写了个很愚蠢的bugaccuracy对应的tensor出来是int型的,我用到了一个除法取平均。而pytorch里无论用 / or // 结果都是取整,,accuracy一直就没有。。所以转换成float就没问题了,,低级bug,下不为例除法的时候要注意数据类型的问题jupyter notebook就会奔溃, 因为torch.tensor是torch.int64数据类型, 而0.137是torch.float32类型.如果改成下面这样, 就会顺利运行

下载地址
用户评论