b9d7f39a752d23a7de60ea6d679812c401fab57e,run.py,,train,#Any#,41

Before Change

        loss = criterion(output, labels)

        loss_list.append(loss.data[0])
        batch_list.append(i+1)

        if i % 10 == 0:
            print("Train - Epoch %d, Batch: %d, Loss: %f" % (epoch, i, loss.data[0]))

After Change

        loss = criterion(output, labels)

        loss_list.append(loss.detach().cpu().item())
        batch_list.append(i+1)

        if i % 10 == 0:
            print("Train - Epoch %d, Batch: %d, Loss: %f" % (epoch, i, loss.detach().cpu().item()))

Instances