How can i use Rotate or pRotate as score in transformer or sentencetransformer module ?

[gsanou]

Hello,
Thank you for your work in mKB, i want to know if i can use rotate or protate in the transformer module, i success to use complex and distmult but cannot do the same for rotate and pRotate. Also I struggle to use other loss function like KLdivergence or BCE. Thank you