From 7aef882f33f5ca180a9a9c11c5aab8ce0f099685 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sat, 22 Jul 2023 23:19:15 +0200 Subject: [PATCH] Update. --- graph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/graph.py b/graph.py index 08f1170..bd80187 100755 --- a/graph.py +++ b/graph.py @@ -20,8 +20,8 @@ def save_attention_image( attention_matrices, # list of 2d tensors T1xT2, T2xT3, ..., Tk-1xTk # do not draw links with a lesser attention min_link_attention=0, - # draw only the strongest links necessary to reache - # min_total_attention + # draw only the strongest links necessary so that their summed + # attention is above min_total_attention min_total_attention=None, # draw only the top k links k_top=None, -- 2.39.5