lightning/tests/core
Nathan Raw 1954d7c87a
Write predictions in LightningModule instead of EvalResult (#3882)
*  add self.write_prediction

*  add self.write_prediction_dict to lightning module
2020-10-05 18:04:02 -04:00
..
__init__.py Refactor model summary + generalize example input array (#1773) 2020-06-15 17:05:58 -04:00
test_datamodules.py ref: bug fix with logging val epoch end + monitor (#3812) 2020-10-03 12:33:29 -04:00
test_decorators.py cleaning tests (#2201) 2020-06-15 22:03:40 -04:00
test_memory.py fix dtype conversion of example_input_array in model summary (#2510) 2020-07-05 07:17:22 -04:00
test_results.py Write predictions in LightningModule instead of EvalResult (#3882) 2020-10-05 18:04:02 -04:00