ac7bb33dd978e2bf6ecb0cf055dd6bf6c9c1ea05,train.py,,,#,21

Before Change

// model.load_weights(opt.weights_path)
model.apply(weights_init_normal)

if cuda:
    model = model.cuda()

model.train()

// Get dataloader
dataloader = torch.utils.data.DataLoader(

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)

        model.seen += imgs.size(0)

        torch.cuda.empty_cache()

Instances