最终查出原因是因为输入网络的数据没有提前定义numpy的array接收,而是直接feed到模型里了,可能这样导致模型一次性保存的太大吧,在feed之前先用np.zeros()定义了数组,用来接收输入的图片数据,再传入network中,问题就能解决了。
tensorflow保存模型时报错: ValueError: GraphDef cannot be larger than 2GB.
最新推荐文章于 2024-04-11 15:42:59 发布
最终查出原因是因为输入网络的数据没有提前定义numpy的array接收,而是直接feed到模型里了,可能这样导致模型一次性保存的太大吧,在feed之前先用np.zeros()定义了数组,用来接收输入的图片数据,再传入network中,问题就能解决了。