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()