94a995defe223eed0898f25d2332ba6178a92abe,ssd/engine/trainer.py,,do_train,#Any#Any#Any#Any#Any#Any#Any#,46

Before Change

                global_step = iteration
                summary_writer.add_scalar("losses/total_loss", losses_reduced.item(), global_step=global_step)
                summary_writer.add_scalar("losses/location_loss", loss_dict_reduced["regression_loss"].item(), global_step=global_step)
                summary_writer.add_scalar("losses/class_loss", loss_dict_reduced["classification_loss"].item(), global_step=global_step)
                summary_writer.add_scalar("lr", optimizer.param_groups[0]["lr"], global_step=global_step)

            tic = time.time()

After Change

            ]
            for loss_name, loss_item in loss_dict_reduced.items():
                log_str.append("{}: {:.3f}".format(loss_name, loss_item.item()))
            log_str = ", ".join(log_str)
            logger.info(log_str)
            if summary_writer:
                global_step = iteration
                summary_writer.add_scalar("losses/total_loss", losses_reduced, global_step=global_step)

Instances