8e9a026bc34d8409faa572a7144c2d96a7c039ed,tests/models/test_restore.py,,test_model_saving_loading,#Any#,411

Before Change

    if not isinstance(dataloaders, list):
        dataloaders = [dataloaders]

    for dataloader in dataloaders:
        for batch in dataloader:
            break

    x, y = batch
    x = x.view(x.size(0), -1)

    // generate preds before saving model

After Change

    if not isinstance(dataloaders, list):
        dataloaders = [dataloaders]

    batch = next(iter(dataloaders[0]))

    // generate preds before saving model
    model.eval()
    pred_before_saving = model(batch)

Instances