lightning_1.3.py

from pytorch_lighting import seed_everything
from pytorch_lightning.callbacks import EarlyStopping
from pytorch_lightning import Trainer

# seeding for torch, numpy, stdlib random, including DataLoader workers!
seed_everything(123, workers=True)

early_stopping = EarlyStopping(
    monitor="val_loss",
    stopping_threshold=1e-4,
    divergence_threshold=9.0,
    check_finite=True,
)

trainer = Trainer(callbacks=[early_stopping],
                  profiler="pytorch",
                  tpu_cores=8)

trainer.validate(model, datamodule=datamodule)
trainer.predict(model, datamodule=datamodule)