[原创] 怎样确认当前正在运行的TensorFlow model-serving服务加载的是哪个.pb模型

跑起来一个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 
感谢关注我的微信公众号(微信扫一扫):
wechat qrcode of codelast
以及我的微信视频号:

发表评论