Excuse me , why the param training=True in eval? If training=True in eval, the mode in handshake will be softmax instead of argmax/activated_selected.

[WangzcBruce]

https://github.com/GT-RIPL/MultiAgentPerception/blob/4ef300547a7f7af2676a034f7cf742b009f57d99/ptsemseg/trainer.py#L968

[drowning-in-codes]

Actually I find that it uses argmax during training in Who2Com which confuses me. And I why the action_argmax is used as a metrics in validation or test but not used during training. I'm also confused that why the dim here is 2 which is not the agent-axis. Have you solved the problem？I hope to get your answer sincerely. https://github.com/GT-RIPL/MultiAgentPerception/blob/4ef300547a7f7af2676a034f7cf742b009f57d99/ptsemseg/models/agent.py#L625C1-L627C45