import precision_tool.tf_config as npu_tf_config config = npu_tf_config.session_dump_config(config, action='dump') sess = tf.Session(config)
import precision_tool.tf_config as npu_tf_config dump_config=npu_tf_config.estimator_dump_config(action='dump') npu_config = NPURunConfig(dump_config=dump_config)
将训练最大步数修改为1后执行训练。Dump数据文件会生成在“precision_data/npu/debug_0/”指定的目录下,即precision_data/npu/debug_0/dump/{time}/{deviceid}/{model_name}/{model_id}/{data_index}目录下。文件目录结构示例:
precision_data/npu/debug_0/dump/20240125153144/0/ge_default_20240125153322_41/6/0/
路径key |
说明 |
备注 |
---|---|---|
dump_path |
dump数据存放路径(如果设置的是相对路径,则为拼接后的全路径)。 |
- |
time |
dump数据文件落盘的时间。 |
格式为:YYYYMMDDHHMMSS |
deviceid |
Device设备ID号。 |
- |
model_name |
子图名称。 |
model_name层可能存在多个文件夹,dump数据取计算图名称对应目录下的数据。 如果model_name出现了“.”、“/”、“\”以及空格时,转换为下划线表示。 |
model_id |
子图ID号。 |
-- |
data_index |
迭代数,用于保存对应迭代的dump数据。 |
如果指定了dump_step,则data_index和dump_step一致;如果不指定dump_step,则data_index序号从0开始计数,每dump一个迭代的数据,序号递增1。 |