变形金刚-如何在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?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)