使用spark-scala调用tensorflow模型

作者: feifeiontheway创建于 2021年5月11日更新于 2021年7月9日

我在单机版上已经运行成功了,但是我在SPARK集群上读取模型时,使用tf.SavedModelBundle.load,显示:无法在指定的导出目录路径中找到SavedModel.pb或.pbtxt文件。请问集群版如何读取模型文件?

内容来源: lyhue1991/eat_tensorflow2_in_30_days