执行完上述步骤后,将在 ${output_dir} 生成以下文件。
名称 | TensorFlow 是否兼容 | 用法 | 描述 |
---|---|---|---|
quantize_train_graph.pb | 是 | 训练 | 量化训练计算图。 |
quantize_eval_graph_{suffix}.pb | 是 | 使用检查点进行评估 | 量化评估计算图,内部冻结有量化信息。该文件有权重,应与检查点文件一起进行评估。 |
quantize_eval_model_{suffix}.pb | 是 |
|
冻结的量化评估计算图,其中冻结有检查点中的权重和量化信息。它可用于在主机上评估量化模型,或者对每一层的输出进行转储以供与 DPU 输出进行交叉比对检查。XIR 编译器使用它作为输入。 |
后缀包含来自检查点文件的迭代信息和日期信息。例如,如果检查点文件为“model.ckpt-2000.*”,而日期为 20200611,那么后缀为“2000_20200611000000”。