动态图加载模型预测相同数据集结果不固定
Created by: xiangyubo
- 版本、环境信息: 1)PaddlePaddle版本:1.5.1 3)GPU:GPU v100 4)系统环境:python 3.x,AI Studio
我自己搭建了一个mobile-net的模型,用于分类。训练了70轮之后,训练集的准确率达到85%,然后我把模型通过 fluid.dygraph.save_persistables 保存。 但是在验证的时候,我通过 fluid.dygraph.load_persistables 加载模型去相同的数据集上验证,准确率很低,而且多次运行时候,准确率还不同。
可以直接在 AI Studio 上 fork 这个项目运行,我已经把训练好的参数保存在当前目录下了,可以不用训练。项目地址: https://aistudio.baidu.com/aistudio/projectdetail/155065