I assume you are using some higher-level API and the Trainer class comes from HuggingFace?
If so, I guess this issue might be related.
I assume you are using some higher-level API and the Trainer class comes from HuggingFace?
If so, I guess this issue might be related.