12 2020 档案
基于tensor2tensor的注意力可视化
摘要:根据训练好的Transformer模型,得到注意力矩阵,并对注意力进行可视化 首先安装:tensorflow 1.13.1 + tensor2tensor 1.13.1 可视化,请在Jupyter notebook中运行。该代码根据tensor2tensor/tensor2tensor/visual
阅读全文
有监督对比损失Tensorflow版本
摘要:这里给出论文的SupContrast: Supervised Contrastive Learning的损失函数Tensorflow版本,代码改自:https://github.com/HobbitLong/SupContrast 损失文件losses.py """ Author: Yonglong
阅读全文