Callback

class paddle.callbacks.Callback

Callback 是一个基类,用于实现用户自定义的callback。

代码示例

  1. import paddle
  2. # build a simple model checkpoint callback
  3. class ModelCheckpoint(paddle.callbacks.Callback):
  4. def __init__(self, save_freq=1, save_dir=None):
  5. self.save_freq = save_freq
  6. self.save_dir = save_dir
  7. def on_epoch_end(self, epoch, logs=None):
  8. if self.model is not None and epoch % self.save_freq == 0:
  9. path = '{}/{}'.format(self.save_dir, epoch)
  10. print('save checkpoint at {}'.format(path))
  11. self.model.save(path)