Text classifier test

TestTextClassifier

Bases: LightningModuleTests[TextClassifier]

Tests for the HF example.

test_overfit_batch

test_overfit_batch(
    algorithm: TextClassifier,
    datamodule: TextClassificationDataModule,
    tmp_path: Path,
    num_steps: int = 3,
)

Test that the loss decreases on a single batch.