projects
/
picoclvr.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[picoclvr.git]
/
graph.py
diff --git
a/graph.py
b/graph.py
index
a819283
..
08f1170
100755
(executable)
--- a/
graph.py
+++ b/
graph.py
@@
-14,16
+14,10
@@
import cairo
def save_attention_image(
def save_attention_image(
- filename,
+ filename,
# image to save
tokens_input,
tokens_output,
tokens_input,
tokens_output,
- # An iterable set of BxHxTxT attention matrices
- attention_matrices,
- pixel_scale=8,
- token_gap=15,
- layer_gap=25,
- y_eps=0.5,
- padding=10,
+ 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
# do not draw links with a lesser attention
min_link_attention=0,
# draw only the strongest links necessary to reache
@@
-32,6
+26,11
@@
def save_attention_image(
# draw only the top k links
k_top=None,
curved=True,
# draw only the top k links
k_top=None,
curved=True,
+ pixel_scale=8,
+ token_gap=15,
+ layer_gap=25,
+ y_eps=0.5,
+ padding=10,
):
if k_top is not None:
am = []
):
if k_top is not None:
am = []