Python mxnet.callback.do_checkpoint用法及代码示例

用法: mxnet.callback.do_checkpoint(prefix, period=1)

参数：

prefix：(str) - 检查点文件名的前缀。
period：(int, optional) - 检查点之间的间隔(时期数)。默认period是 1。

callback：- 一个回调函数，可以作为epoch_end_callback适合。

返回类型：

函数

每隔几个 epoch 保存一个模型检查点的回调。每个检查点由几个二进制文件组成：一个模型说明文件和一个参数(权重和偏差)文件。模型说明文件命名为prefix-symbol.json，参数文件命名为prefix-epoch_number.params

示例：

>>> module.fit(iterator, num_epoch=n_epoch,
... epoch_end_callback  = mx.callback.do_checkpoint("mymodel", 1))
Start training with [cpu(0)]
Epoch[0] Resetting Data Iterator
Epoch[0] Time cost=0.100
Saved checkpoint to "mymodel-0001.params"
Epoch[1] Resetting Data Iterator
Epoch[1] Time cost=0.060
Saved checkpoint to "mymodel-0002.params"

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.callback.do_checkpoint。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

返回类型：

示例：