tensorflow(3)— 常用命令汇总(打印、Pb保存等)

1、打印tfrecord,查看样本数据类型

import tensorflow as tf
import os
import glob
count=0
path1='/Users/wangruojing/mt_project/mix_rerank/inputs/part-r-00002'

for example in tf.python_io.tf_record_iterator(path1):

  a = tf.train.Example.FromString(example)
  count += 1
  print(a)
  #print(a["total_price"])

  if count>1: break

2、报错:DataLossError: corrupted record at 0

  1. 报错:tensorflow.python.framework.errors_impl.DataLossError: corrupted record at 0
  2. 解决方式:ls -al,删除文件下的掩藏文件即可

3、查看输入输出是否正确

tf模型保存为 pb 后,验证输入输出是否正确

saved_model_cli show --all --dir ./outputs/intelli_1009_1

 

posted @ 2021-10-13 11:32  深度机器学习  阅读(220)  评论(0编辑  收藏  举报