OSError Unable to load weights from pytorch checkpoint file. If you tried to load a PyTorch model from a TF 2.0 checkpoint, please set from_tf=True
时间: 2023-08-15 20:08:55 浏览: 354
解决Pytorch 加载训练好的模型 遇到的error问题
5星 · 资源好评率100%
这个错误通常表示你试图从一个不兼容的 PyTorch checkpoint 文件中加载权重。如果你是从 TensorFlow 2.0 的 checkpoint 文件中加载 PyTorch 模型,请确保将参数 `from_tf=True` 设置为 True。如果这不是你的情况,请确保你正在尝试加载一个与你的 PyTorch 版本兼容的 checkpoint 文件,并且没有其他错误导致了这个问题。
阅读全文