【承接图像分类、检测、分割、生成相关项目,私信。】
【MXNet Gluon】自己动手实现断点保存
…用过caffe的炼丹师应该都知道,在用caffe训练模型时,可以通过命令行窗口提前终止训练过程,caffe会自动保存当前状态的参数,以供继续训练。
…但是,对于MXNet,无论你使用何种接口,都不存在这种机制。
…在这里,提供一个实现方式。
实现该过程,分两个步骤:
…一是写自己的训练过程函数
…二是监控模型训练过程中来自命令行的按键相应
实现如下:
Gluon接口
'''
相关头文件
'''
import signal
def