ddp pickle

This commit is contained in:
William Falcon
2020-04-27 08:04:02 -04:00
parent a3cebb4469
commit 710dbf5a12
+1 -1
View File
@@ -30,7 +30,7 @@ from tests.base import TestModelBase
def test_model_pickle(tmpdir):
import pickle
model = TestModelBase()
model = TestModelBase(tutils.get_default_hparams())
pickle.dumps(model)