21f9151d745c57671661baff3272d544a56f7bbe,trainval_net.py,,,#,148

Before Change

          loss_rpn_cls = rpn_loss_cls.data[0]
          loss_rpn_box = rpn_loss_box.data[0]
          loss_rcnn_cls = RCNN_loss_cls.data[0]
          loss_rcnn_box = RCNN_loss_bbox.data[0]
          fg_cnt = torch.sum(rois_label.data.ne(0))
          bg_cnt = rois_label.data.numel() - fg_cnt

        print("[session %d][epoch %2d][iter %4d/%4d] loss: %.4f, lr: %.2e" \

After Change

    print(end - start)

    if args.use_tfboard:
      logger.close()

Instances