Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
lightning/tests/base/__init__.py

7 lines
249 B
Python
Raw Normal View History

CI: split tests-examples (#990) * CI: split tests-examples * tests without template * comment depends * CircleCI typo * add doctest * update test req. * CI tests * setup macOS * longer train * lover pred acc * fix model * rename default model * lower tests acc * typo * imports * fix test optimizer * update calls * fix Win * lower Drone image * fix call * pytorch image * fix test * add dev image * add dev image * update image * drone volume * lint * update test notes * rename tests/models >> tests/base * group models * conftest * optim imports * typos * fix import * fix tests * install AMP * tests * fix import
2020-03-25 11:46:27 +00:00
"""Models for testing."""
Tests: refactor cleanup (#1744) * wip * cleaning * optim imports * - * default hparams * fix restore * fix imports
2020-05-10 17:15:28 +00:00
from tests.base.datasets import TrialMNIST
More robust way of collecting init argument names for LightningModules (#3066) When a LightningModule inherits from a class that implements `__new__()` such as `typing.Generic`, `inspect.signature(cls)` short-circuits and returns the signature of `__new__()` instead of `__init__()`. So, we need to be more specific and call inspection directly on the init function.
2020-08-20 11:19:11 +00:00
from tests.base.model_template import EvalModelTemplate, GenericEvalModelTemplate
ref: bug fix with logging val epoch end + monitor (#3812) * ref: fix metric err * ref: fix metric err * ref: fix metric err * ref: merge * ref: merge * ref: merge * ref: merge * ref: decoupled ddp2 * ref: decoupled ddp2 * ref: decoupled ddp2 * ref: decoupled ddp2 * ref: decoupled ddp2 * ref: clean up ddp before final fix * ref: clean up ddp before final fix * ref: clean up ddp before final fix * ref: clean up ddp before final fix * ref: clean up ddp before final fix * ref: clean up ddp before final fix * ref: clean up ddp before final fix * ref: clean up ddp before final fix * ref: clean up ddp before final fix * ref: clean up ddp before final fix * ref: clean up ddp before final fix
2020-10-03 16:33:29 +00:00
from tests.base.simple_model import SimpleModule
verified epoch logging (#3830) * ref: fix epoch logging * verified epoch logging * verified epoch logging * verified epoch logging * verified epoch logging * verified epoch logging * verified epoch logging * verified epoch logging * verified epoch logging
2020-10-04 01:17:24 +00:00
from tests.base.boring_model import BoringModel