变形金刚-如何在Torch版本1.4.0Albert中加载pytorch_model.bin

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

如果我安装了torch == 1.6.0，则可以成功加载模型。但是，我必须使用Torch版本1.4.0和Torchvision 0.5.0。如何在火炬1.4.0中加载pytorch_model.bin ？？

++我曾尝试在割炬1.4.0版中对run_language_modeling.py进行训练，但它无法导入“ torch.optim.lr_scheduler”，因此无法执行训练代码。

所以我的问题是 [1]如何在火炬版本1.4.0中加载pytorch_model.bin /或 [2]如何在炬管版本1.4.0中训练run_language_modeling.py？