7fffa98b9166a03b4a53fb40202d97b09e8e9036,train.py,,,#,21

Before Change

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

        loss = model(imgs, targets)

        loss.backward()
        optimizer.step()

After 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:

Instances