lightning/tests/trainer
Rohit Gupta ac76dfcf62
Remove NaNs from loss in LRFinder (#1862)
* Remove NaNs from loss in LRFinder

* np.isfinite

* chlog

* add test

* chlog

Co-authored-by: Jirka <jirka@pytorchlightning.ai>
2020-05-19 08:39:19 +02:00
..
__init__.py split trainer tests (#956) 2020-02-26 20:31:40 -05:00
test_checks.py add test for trainer.test() (#1858) 2020-05-17 16:30:20 -04:00
test_dataloaders.py remove extra kwargs from Trainer init (#1820) 2020-05-17 09:14:54 -04:00
test_lr_finder.py Remove NaNs from loss in LRFinder (#1862) 2020-05-19 08:39:19 +02:00
test_optimizers.py Tests: refactor cleanup (#1744) 2020-05-10 13:15:28 -04:00
test_trainer.py Fix `save_weights_only` flag in ModelCheckpoint (#1780) 2020-05-17 09:24:17 -04:00
test_trainer_cli.py extend arg parser (#1842) 2020-05-14 17:56:11 -04:00
test_trainer_tricks.py Tests: refactor cleanup (#1744) 2020-05-10 13:15:28 -04:00