跑起来一个TensorFlow model-serving服务后,有时候记不清它加载的是哪个.pb模型了,可以采用下面的办法来确认。
访问URL:
http://<your_model_serving_host>:18501/v1/models/<your_model_name>
其中:
<your_model_serving_host> 是你的model-serving服务器的域名或IP。
<your_model_name> 是你的模型名称。
文章来源:https://www.codelast.com/
会看到页面输出类似于下面的内容:
{"model_version_status": [{"version": "1684833957","state": "AVAILABLE","status": {"error_code": "OK","error_message": ""}}]}
其中,version就是我们要找的东西。
到你保存.pb模型的父目录下(可能是HDFS或本地磁盘),无脑搜version对应的关键字 1684833957,找到哪个目录,就是我们要找的.pb模型所在的目录。
通常这个目录下会有一个 saved_model.pb 文件,以及一个 variables 子目录。
为什么可以这样做?因为version里的时间戳就是导出 .pb 模型的时间戳,这个时间戳精确到秒,一般情况下,两个模型几乎不太可能在同一秒生成,所以这个时间戳是唯一的,因此只要能找到这个目录名,那么目录里的 .pb 模型几乎肯定是我们要找的模型。
文章来源:https://www.codelast.com/
➤➤ 版权声明 ➤➤
转载需注明出处:codelast.com
感谢关注我的微信公众号(微信扫一扫):
以及我的微信视频号: