模型加载

  统计/机器学习 计算机视觉    浏览次数:605        分享
0
请问怎么在测试时一次性加载好模型参数并初始化,防止后面在测试时每次都加载模型
 

klme   2021-05-18 22:00



   1个回答 
0

保存的代码:

torch.save(model.state_dict(), PATH)

加载的代码:

model = TheModelClass(*args, **kwargs)

model.load_state_dict(torch.load(PATH))


SofaSofa数据科学社区DS面试题库 DS面经

飞时过   2021-09-22 13:55



  相关主题

图像处理中rgba是什么意思?   2回答

coco数据集是什么?   2回答

有什么好的方法识别图片中的特殊点呢   1回答

关于video saliency detection的一些问题   0回答

cifar-10数据集是有哪十类?   1回答

pip install opencv-python失败,难道非要编译安装?   3回答

Python中怎么把一个RGB彩色图像转成灰度黑白图像?   3回答

opencv里waitkey和destroyAllWindows有什么用?   2回答

图像轮廓识别   1回答

ubuntu系统中怎么安装tesseract-ocr的中文识别库?   1回答

怎么用python读取视频文件?   2回答

对图像进行数据增广有哪些方法?   2回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!