Skip to content

travel-go/pytorch-attn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pytorch-attn

draw_attn

注:暂时只能用于画基于Tenosr的attn权重,文本文件的attn之后更新

数据:

attn:用pytorch的torch.save把attn权重保存下

source_sentence:源端的句子,长度需要和attn中src大小一致

target_sentence:源端的句子,长度需要和attn中tgt大小一致

用法: python draw_attn.py -attn_path $attn_path -save_dir $save_dir -source_file $source_file -target_file $target_file

-attn_path是attn文件的路径

-source_file是源端句子文件的路径

-target_file是目标端句子文件的路径

-save_dir是存储的文件夹路径

-plt_show是否在窗口输出attn图片,默认是输出,也可以设置为0,不输出。如果是服务器上,就设置为0(服务器上无法显示图片)

-cbar:是否显示attn图片右侧的bar

Releases

No releases published

Packages

No packages published

Languages