wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactored model tests

Browse Source
This commit is contained in:
William Falcon
2019-07-24 12:14:26 -04:00
parent c7ad04be57
commit 24ceafa05c
1 changed files with 2 additions and 2 deletions
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)