From 18dff98549e2b8a9b961e90c7cf4d0715f44b39b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Fri, 15 May 2020 20:39:48 +0200 Subject: [PATCH] fix test --- tests/models/test_hooks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/test_hooks.py b/tests/models/test_hooks.py index 88abe8f048438..8537b90ff6d4c 100644 --- a/tests/models/test_hooks.py +++ b/tests/models/test_hooks.py @@ -49,7 +49,7 @@ def transfer_batch_to_device(self, batch, device): batch.targets = batch.targets.to(device) return batch - model = CurrentTestModel(tutils.get_default_hparams()) + model = CurrentTestModel() batch = CustomBatch((torch.zeros(5, 28), torch.ones(5, 1, dtype=torch.long))) trainer = Trainer()