Method test_mnist

tests/test_pytorch_lightning.py:65–72

(self)

        dm = LitDataModule()
        model = LitClassifier()
        trainer = pl.Trainer(max_epochs=1, log_every_n_steps=1, devices=1)
        trainer.fit(model, datamodule=dm)

        self.assertIn("train_loss", trainer.logged_metrics)
        self.assertIn("val_loss", trainer.logged_metrics)