bc4dfbcc2ecae4b3a4bedc2bd18909a9208c05cb,train.py,,,#,21

Before Change

        targets = Variable(targets.type(Tensor), requires_grad=False)

        loss = 0
        for i in range(subdivisions):
            optimizer.zero_grad()
            sub_imgs = imgs[i*sub_batch: (i+1)*sub_batch]
            sub_targets = targets[i*sub_batch: (i+1)*sub_batch]

            loss += model(sub_imgs, sub_targets)

        loss.backward()
        optimizer.step()

    if epoch % opt.checkpoint_interval == 0:

After Change

// Initiate model
model = Darknet(opt.model_config_path)
model.apply(weights_init_normal)

if cuda:
    model = model.cuda()

Instances