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

refactored model tests

Browse Source
This commit is contained in:
William Falcon committed 2019-07-24 12:14:26 -04:00
1 parent c7ad04be57
commit 24ceafa05c
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+2 -2
View File
@@ -184,8 +184,8 @@ def run_gpu_model_test(trainer_options):
checkpoint = ModelCheckpoint(save_dir)
# add these to the trainer options
trainer_options.checkpoint_callback = checkpoint
trainer_options.experiment = exp
trainer_options['checkpoint_callback'] = checkpoint
trainer_options['experiment'] = exp
# fit model
trainer = Trainer(**trainer_options)