diff --git a/notebooks/quantized_recurrent.ipynb b/notebooks/quantized_recurrent.ipynb index 5d28c0bb9..766e82745 100644 --- a/notebooks/quantized_recurrent.ipynb +++ b/notebooks/quantized_recurrent.ipynb @@ -649,7 +649,6 @@ "\n", "# Generate random input\n", "inp = torch.randn(5, 2, 10)\n", - "torch.allclose\n", "# Check outputs are the same\n", "assert torch.allclose(quant_rnn(inp)[0], float_rnn(inp)[0], atol=ATOL)\n", "# Check hidden states are the same\n",