einsum(): the number of subscripts in the equation (3) does not match the number of dimensions (2) for operand 1 and no ellipsis was given #8
Description
Encounter this error while try the train.py with model KTransCAT, and this issue may be inside the layer FeatTransKCat. Also may I ask the FGN in the paper refer to which model?