文件名称：attend-and-compare
文件大小：1.67MB
文件格式：ZIP
更新时间：2024-05-24 13:05:37
Python
参加并比较 这是论文的正式实现。 用法 import torch from context_module import ACM x1 = torch . randn ( 256 * 20 * 20 * 5 ). view ( 5 , 256 , 20 , 20 ). float () acm = ACM ( num_heads = 32 , num_features = 256 ) acm . init_parameters () y , dp = acm ( x1 ) print ( y . shape ) # output print ( dp . shape ) # dot product for orthogonal loss 例子 以下目录中提供了CXR14和COCO实验代码。 引文 Minchul Kim *，Jongchan Park *，Seil Na，Chang M