ac7bb33dd978e2bf6ecb0cf055dd6bf6c9c1ea05,train.py,,,#,21

Before Change

        model.seen += imgs.size(0)

    if epoch % opt.checkpoint_interval == 0:
        model.save_weights("%s/%d.weights" % (opt.checkpoint_dir, epoch))

After Change

            )
        )

        batches_done = len(dataloader) * epoch + batch_i

        // Tensorboard logging
        for loss_name, loss in model.losses.items():
            logger.scalar_summary(loss_name, loss, batches_done)

Instances