slei109/PATNet

Question about the TAFT module

Closed this issue · 1 comments

Thanks for your work!
A regression loss is used in TAFT for training the transformation matrix. But there is no such loss in your work.
Does the PATM module not need to be trained? Why?

PATM is trained. We adopt Cross_Entropy as the loss function and implement it in the compute_objective() function.