From edd295b38648e0e250b26eee394b4c84507fe242 Mon Sep 17 00:00:00 2001 From: jloveric Date: Sun, 26 Nov 2023 17:07:33 -0800 Subject: [PATCH] minor --- tests/test_attention_network.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_attention_network.py b/tests/test_attention_network.py index cf73c12..c70c0ce 100644 --- 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