Solving test CAN (looping)

pyt-team · Apr 4, 2024 · db96096 · db96096
1 parent bf30e09
commit db96096
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/nn/cell/test_can.py b/test/nn/cell/test_can.py
@@ -20,6 +20,7 @@ def test_forward(self):
             heads=1,
             n_layers=2,
             att_lift=False,
+            pooling=True,
         ).to(device)
 
         x_0 = torch.rand(2, 2)