ddp pickle

williamFalcon · williamFalcon · commit 710dbf5a1280 · 2020-04-27T08:04:02.000-04:00
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)
 
 
