Deep learning 变压器-如何在torch版本1.4.0（Albert）中加载pytorch_model.bin_Deep Learning_Bert Language Model_Huggingface Transformers_Language Model

Deep learning 变压器-如何在torch版本1.4.0（Albert）中加载pytorch_model.bin

deep-learning

Deep learning 变压器-如何在torch版本1.4.0（Albert）中加载pytorch_model.bin,deep-learning,bert-language-model,huggingface-transformers,language-model,Deep Learning,Bert Language Model,Huggingface Transformers,Language Model,火炬1.4.0版我执行run_language_modeling.py并保存模型。但是，当我加载保存的模型时，会出现“OSError:无法从pytorch检查点文件加载权重。如果您试图从TF 2.0检查点加载pytorch模型，请从_TF=True设置” 如果我安装torch==1.6.0，则加载模型是成功的。但是，我必须使用torch版本1.4.0和torchvision 0.5.0。如何在torch 1.4.0中加载pytorch_model.bin ++我试图在torch版本1.4.0中

火炬1.4.0版我执行run_language_modeling.py并保存模型。但是，当我加载保存的模型时，会出现“OSError:无法从pytorch检查点文件加载权重。如果您试图从TF 2.0检查点加载pytorch模型，请从_TF=True设置”

如果我安装torch==1.6.0，则加载模型是成功的。但是，我必须使用torch版本1.4.0和torchvision 0.5.0。如何在torch 1.4.0中加载pytorch_model.bin

++我试图在torch版本1.4.0中训练run_language_modeling.py，但它无法导入“torch.optim.lr_scheduler”，因此训练代码无法执行

因此，我的问题是 [1] 如何在torch版本1.4.0/或中加载pytorch_model.bin [2] 如何在torch版本1.4.0中培训run_language_modeling.py