wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-29 13:30:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

add test for trainer.test() (#1858)

Browse Source 
* fix trainer.test()

* Update trainer.py

Co-authored-by: William Falcon <waf2107@columbia.edu>
This commit is contained in:
Victor QuachandWilliam Falcon authored and GitHub committed 2020-05-17 16:30:20 -04:00
1 parent d7f9c03663
commit 1a797bdad5
1 file changed
+11
Download Patch File Download Diff File Expand all files Collapse all files
tests/trainer/test_checks.py
+11
View File
@@ -121,3 +121,14 @@ def test_wrong_test_settigs(tmpdir):
model.test_dataloader = LightningModule.test_dataloader
model.test_epoch_end = None
trainer.test(model, test_dataloaders=model.dataloader(train=False))
# ----------------
# if we are just testing, no need for train_dataloader, train_step, val_dataloader, and val_step
# ----------------
model = EvalModelTemplate(hparams)
model.test_dataloader = LightningModule.test_dataloader
model.train_dataloader = None
model.train_step = None
model.val_dataloader = None
model.val_step = None
trainer.test(model, test_dataloaders=model.dataloader(train=False))