[Bug]: key error in weight_dict & losses
zhiyuanyou opened this issue · 1 comments
zhiyuanyou commented
In models/p2pnet.py
:
Line 278: losses['loss_point'] = loss_bbox.sum() / num_points
Line 335: weight_dict = {'loss_ce': 1, 'loss_points': args.point_loss_coef}
one is "loss_points" and another is "loss_point" (an "s" is different) ! This would cause that the point loss is actually not used, which is fully wrong!
dddddasda commented
I found that too, but I can get results comparable with the official ones on ShanghaiTech A dataset even without point loss item. How about you?