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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

返回類型：

示例：