amingwu/ccn

Connective Cognition Network for Directional Visual Commonsense Reasoning

ccn's Issues

晚辈请教您几个问题

前辈你好，我是研一的学生，研究方向为visual storytelling,
我目前在重写《No Metrics Are Perfect:Adversarial Reward Learning for Visual Storytelling》，遇到了很多问题。
我看见您在原作者的GitHub下面留言了，如果可以的话，我们可以给个联系方式（邮箱也可以），我想您请教几个问题吗？
因为我们实验室没人研究这个，我属于摸着石头过河，希望前辈给个机会，谢谢您。
我的邮箱是[email protected]
我的vx是13093193765
我的QQ是841222073
万分感谢，谢谢您，祝您生活愉快！！！！！！！！！！

请问softmax的维度这里为什么不是2或者-1

CCN/train/train.py

Line 242 in 2254049

object_graph = masked_softmax(object_graph, diag_zero[None,...].cuda(), dim=1) + diag[None,...].cuda()

你好，按照论文里写的这里softmax是对每一行进行操作，obj_graph的维度是三维的，我的理解应该是对最后一维进行softmax，dim的设置应该是2或者-1吧？