Text classification example test
TestTextClassificationExample #
Bases: LearningAlgorithmTests[TextClassificationExample]
Tests for the HF example.
test_overfit_batch #
test_overfit_batch(
algorithm: TextClassificationExample,
datamodule: TextClassificationDataModule,
tmp_path: Path,
num_steps: int = 3,
)
Test that the loss decreases on a single batch.