ddp pickle

Lightning-AI · Apr 27, 2020 · 710dbf5 · 710dbf5
1 parent a3cebb4
commit 710dbf5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
@@ -30,7 +30,7 @@
 def test_model_pickle(tmpdir):
     import pickle
 
-    model = TestModelBase()
+    model = TestModelBase(tutils.get_default_hparams())
     pickle.dumps(model)