mmdet复现遇到的问题——UserWarning: floordiv is deprecated

SupCodeCat

已于 2024-05-03 16:45:36 修改

阅读量1k

点赞数 7

分类专栏： mmdetection 文章标签： mmdetection python pytorch

于 2024-03-12 11:12:05 首次发布

本文链接：https://blog.csdn.net/struggle_success/article/details/136645642

版权

mmdetection 专栏收录该内容

4 篇文章

订阅专栏

文章讲述了在PyTorch中遇到关于__floordiv__函数行为改变的UserWarning，介绍了如何通过torch.div函数并设置rounding_mode参数来解决此问题，推荐使用trunc或floor模式进行正确取整。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、报错信息

UserWarning: __floordiv__ is deprecated, and its behavior will change in a future version of 
pytorch. It currently rounds toward 0 (like the 'trunc' function NOT 'floor'). 
This results in incorrect rounding for negative values. To keep the current behavior, use 
torch.div(a, b, rounding_mode='trunc'), or for actual floor division, 
use torch.div(a, b, rounding_mode='floor'). 
scales = 0.5 + (areas - min_area) // (max_area - min_area)

二、错误分析

大概意思就是向下取整运算符 // 已经被废弃，使用torch.div(a,b,rounding_mode=‘floor’)或者torch.div(a,b,rounding_mode=‘trunc’)

torch.div(a,b,rounding_mode='floor')  ## 向下取整
torch.div(a,b,rounding_mode='trunc')  ## 四舍五入

三、解决问题

scales = 0.5 + (areas - min_area) // (max_area - min_area)
# 改成
scales = 0.5 + torch.div((areas - min_area) , (max_area - min_area) , rounding_mode='floor')