115a1c5f642c0285afc46d0b49eafb4b192a0afa,train.py,,train_model,#Any#Any#Any#Any#Any#,142

Before Change

                running_corrects += torch.sum(preds == labels.data)

            epoch_loss = running_loss / dataset_sizes[phase]
            if int(version[2]) > 3: // for the new version like 0.4.0 and 0.5.0
                running_corrects = running_corrects.item()
            epoch_acc = running_corrects / dataset_sizes[phase]
            
            print("{} Loss: {:.4f} Acc: {:.4f}".format(
                phase, epoch_loss, epoch_acc))

After Change

                    running_loss += loss.item()
                else :  // for the old version like 0.3.0 and 0.3.1
                    running_loss += loss.data[0]
                running_corrects += float(torch.sum(preds == labels.data))

            epoch_loss = running_loss / dataset_sizes[phase]
            epoch_acc = running_corrects / dataset_sizes[phase]
            

Instances