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类对象。

调用示例

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])