detectron2保存模型由pth转pt

detectron2保存模型由pth转pt checkpoint源代码: https://detectron2.readthedocs.io/en/latest/_modules/fvcore/common/checkpoint.html#Checkpointer
重写from fvcore.common.checkpoint import Checkpointer中的Checkpointer类中的save方法
class MYCheckpointer(Checkpointer):
def save(self, name, **kwargs):
if not self.save_dir or not self.save_to_disk:
return

data = {}
data["model"] = self.model.state_dict()

detectron2保存模型由pth转pt最先出现在Python成神之路

版权声明:
作者:cc
链接:https://www.techfm.club/p/29583.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>