lightning/tests/base/__init__.py

5 lines
152 B
Python

"""Models for testing."""
from tests.base.datasets import TrialMNIST
from tests.base.model_template import EvalModelTemplate, GenericEvalModelTemplate