test problem

[wzylove]

I encountered this problem during testing and I don’t know how to solve it.

[youyoubud]

请问pytorch-lightning版本是什么啊？我总是冲突

[jinsuyoo]

Thanks for your interest in our work!

It seems related to pytorch-lightning version issue, but I don't remember which version I used for the time of research..

Here is a link for a similar problem and solution that may help to solve the issue:
Lightning-AI/pytorch-lightning#2909

You may try different lightning versions or modify the model loading code as the link suggests.

Hope this will help!!

Best,
Jinsu