0a7220ed201eee2ad463264a72d5ce42c33dbcc1,trainval_net.py,,,#,113

Before Change

    os.makedirs(output_dir)

  dataset = roibatchLoader(roidb, imdb.num_classes, training=False,
                        normalize = transforms.Normalize(
                        mean=[0.485, 0.456, 0.406],
                        std=[0.229, 0.224, 0.225]))

  dataloader = torch.utils.data.DataLoader(dataset, batch_size=1,
                            shuffle=True, num_workers=0)

After Change

             fasterRCNN.RCNN_loss_cls.data[0], \
             fasterRCNN.RCNN_loss_bbox.data[0]))
          if args.use_tfboard:
            info = {
              "loss": loss_temp / args.disp_interval,
              "loss_rpn_cls": fasterRCNN.RCNN_base.RCNN_rpn.rpn_loss_cls.data[0],
              "loss_rpn_box": fasterRCNN.RCNN_base.RCNN_rpn.rpn_loss_box.data[0],
              "loss_rcnn_cls": fasterRCNN.RCNN_loss_cls.data[0],
              "loss_rcnn_box": fasterRCNN.RCNN_loss_bbox.data[0]
            }
            for tag, value in info.items():
              logger.scalar_summary(tag, value, step)

        loss_temp = 0

Instances