SetFit customized model head got " maximum recursion depth exceeded" errors

Beginners
1

Hi, I am trying to customize my own SetFitHead with the scripts as below:

Screenshot 2024-09-09 at 11.43.46
Screenshot 2024-09-09 at 11.43.461912×1178 322 KB

. I have run smoothly with setfit trainer but today I suddenly got
" maximum recursion depth exceeded" errors. If I exchange the head as SetFitHead then it works again. Can anyone help with this ?

2

The error happens in when I call “trainer.train()”