CasRel-pytorch-reimplement framework.py中为什么保存ori

framework.py中为什么保存ori_model而不是model

Open senseandcalm opened this issue 3 years ago • 1 comments

framework.py第120行，保存的是ori_model而不是model？ ori_model不是最原始的模型么？

Mar 18 '21 09:03 senseandcalm

因为你可以用多卡训练，多卡训练的情况下model是多卡，因为我不想保存多卡的模型，所以保存的是ori_model，两者在单卡训练时是一致的，多卡模型的保存与加载可以参考https://my.oschina.net/u/4279077/blog/3366893

Mar 19 '21 08:03 longlongman