031392ff2cbb5703558d17a676a456521f754190,mnist/main.py,,train,#Any#,67

Before Change

        optimizer.zero_grad()
        batch_data.data[:] = training_data[i:i+BATCH_SIZE]
        batch_targets.data[:] = training_labels[i:i+BATCH_SIZE]
        loss = criterion(model(batch_data), batch_targets)
        loss.backward()
        loss = loss.data[0]
        optimizer.step()
        print("Epoch: {} [{}/{} ({:.0f}%)]\tLoss: {:.4f}".format(epoch,

After Change

        batch_data.data[:] = training_data[start:end]
        batch_targets.data[:] = training_labels[start:end]
        output = model(batch_data)
        loss = criterion(output, batch_targets)
        loss.backward()
        loss = loss.data[0]
        optimizer.step()
        print("Train Epoch: {} [{}/{} ({:.0f}%)]\tLoss: {:.4f}"

Instances