minor

jloveric · Nov 27, 2023 · edd295b · edd295b
1 parent b8ca58a
commit edd295b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_attention_network.py b/tests/test_attention_network.py
@@ -40,6 +40,7 @@ def test_attention_network():
         heads =2
     )
     result = network(input_data)
+    print('result', result)
     print("result", result.shape)
     assert result.shape[0] == 32
     assert result.shape[1] == 128