marui_955

Hello, I also find this problem, my resolution is change the code 'if boxes.numel()==0 return keep' to '... ... return keep, 0'. Because the count means the kept boxes number,...

Are you sure that the Loss Model is in the training state? Auto-grad-discrese should also apply to the loss model !!!