X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=graph.py;h=bd801875bea7531fb0a480d609ff408f887956d6;hb=7aef882f33f5ca180a9a9c11c5aab8ce0f099685;hp=08f1170b9dfc4838340e972d8731ace834967246;hpb=95717a8bf88159051f9c4b8862b0b643187826e9;p=picoclvr.git

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,