pip3 install gnureadline pexpect
import precision_tool.tf_config as npu_tf_config sess = npu_tf_config.sess_dump(sess=sess)
import precision_tool.tf_config as npu_tf_config estim_specs = tf.estimator.EstimatorSpec(training_hooks=[npu_tf_config.estimator_dump()])
将训练最大步数修改为1后执行训练,会在“precision_data/tf/tf_debug/”目录生成dump数据。
执行python3 precision_tool/cli.py tf_dump后,会在“precision_data/tf/dump/”目录生成解析好的dump数据。如果需要重新生成dump数据,将已生成的数据删除再重新执行训练和解析操作即可。