NPUCheckpointSaverHook构造函数

函数原型

def __init__(self,checkpoint_dir,save_secs=None,save_steps=None,saver=None,checkpoint_basename="model.ckpt",scaffold=None,listeners=None)

功能说明

NPUCheckpointSaverHook类的构造函数，用于保存checkpoint文件。NPUCheckpointSaverHook类继承了CheckpointSaverHook类，可以调用基类的原生接口，用于记录训练过程中的模型信息。

使用约束

在使用NPUEstimator并且配置iteration_per_loop>1时，该Hook可能不生效。

参数说明

返回值

返回NPUCheckpointSaverHook类对象。

调用示例

1
2
3
4
5
6
7

from npu_bridge.npu_init import *
checkpoint_hook = NPUCheckpointSaverHook(checkpoint_dir='./ckpt', save_steps=2000)
...
mnist_classifier.train(   
   input_fn=train_input_fn,
    steps=2000,
    hooks=[checkpoint_hook])