1aac481957dda7b9873a4204b5c96cb4e2c205c6,tests/models/test_tpu.py,,test_model_tpu_early_stop,#Any#,178

Before Change

        max_epochs=2,
        limit_train_batches=2,
        limit_val_batches=2,
        tpu_cores=[1],
    )
    trainer.fit(model)

After Change

        tpu_cores=8,
    )
    trainer.fit(model)
    trainer.test(test_dataloaders=DataLoader(RandomDataset(32, 2000), batch_size=32))


@RunIf(tpu=True)

Instances