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

            if summary_writer:
                global_step = iteration
                summary_writer.add_scalar("losses/total_loss", losses_reduced, global_step=global_step)
                for loss_name, loss_item in loss_dict_reduced.items():
                    summary_writer.add_scalar("losses/{}".format(loss_name), loss_item, global_step=global_step)
                summary_writer.add_scalar("lr", optimizer.param_groups[0]["lr"], global_step=global_step)

            tic = time.time()

Instances