Closed StephenZhao1 closed 1 day ago
It appears to be related to using a newer version of PyTorch. A similar problem has been discussed in this issue.
While there may be a more detailed solution in the discussion thread, the simplest workaround is to downgrade your PyTorch version:
This should resolve the compatibility issue you're experiencing.
with out modify any code,train on first epoch will meet this problem even change batch size to 2