bubbliiiing/retinanet-pytorch

计算fps值报错

Vegetabhl opened this issue · 1 comments

您好,我用您的代码训练完,也测得了对应的map指标。但是在计算fps值的时候报了如下错误 TypeError: non_max_suppression() got multiple values for argument 'conf_thres',麻烦请问下您有遇到过这个错误吗? 我的环境是ubuntu下torch1.8,谢谢了。

多传了个参数,已修改