Adrian Wälchli 02fa32b7bc Handle torch.jit scripted modules in layer summary (#6511)		2021-03-15 03:17:42 +01:00
..
__init__.py	Refactor model summary + generalize example input array (#1773)	2020-06-15 17:05:58 -04:00
test_datamodules.py	Add Trainer.validate(…) method to run one validation epoch (#4948)	2021-03-11 03:46:37 +01:00
test_decorators.py	Refactor: skipif for Windows 2/n (#6268)	2021-03-02 09:36:01 +00:00
test_lightning_module.py	Call `optimizer.zero_grad()` before backward inside closure in AutoOpt (#6147)	2021-03-01 14:36:46 +01:00
test_lightning_optimizer.py	Remove optimizer_idx arg in manual optimization (#6093)	2021-03-07 08:48:50 +01:00
test_memory.py	Handle torch.jit scripted modules in layer summary (#6511)	2021-03-15 03:17:42 +01:00
test_metric_result_integration.py	Refactor: skipif for Windows 2/n (#6268)	2021-03-02 09:36:01 +00:00
test_results.py	Typing for tests 1/n (#6313)	2021-03-09 11:27:15 +00:00