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

Before Change

  if args.optimizer == "adam":
    lr = lr * 0.1
    optimizer = torch.optim.Adam([
      {"params": fasterRCNN.RCNN_base.RCNN_base_model[0].parameters(), "lr": lr * 0.0},
      {"params": fasterRCNN.RCNN_base.RCNN_base_model[1].parameters(), "lr": lr * 0.1},
      {"params": fasterRCNN.RCNN_base.RCNN_base_model[2].parameters()},
      {"params": fasterRCNN.RCNN_base.RCNN_rpn.parameters()},

After Change

              "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

      if (step % args.checkpoint_interval == 0) and step > 0:
        //   pdb.set_trace()

Instances