A RuntimeError about pointaugment

[Jcxloyal]

hello friend, I met the same problem, I solved it by rearranging the code in the model.py

hope it helps

[ReunionJ]

hello friend, I met the same problem, I solved it by rearranging the code in the model.py

hope it helps

Thank you very much for your reply!
But after modifying the code, I encountered the situation that the accuracy of model prediction is very low. I don't know if it is because the epoch is too small.
Do you encounter the same situation?

[Jcxloyal]

hello friend, I met the same problem, I solved it by rearranging the code in the model.py
hope it helps

Thank you very much for your reply!
But after modifying the code, I encountered the situation that the accuracy of model prediction is very low. I don't know if it is because the epoch is too small.
Do you encounter the same situation?

I'm sorry, friend. I didn't encounter the situation

[sunhufei]

您好，这个程序可以把你改过的代码发给我一份ma？