train_util.py 文件源码

def save_params(prefix, mdl, opt):
  logger = logging.getLogger(__name__)

  logger.info('Saving model/optimizer parameters')
  chainer.serializers.save_npz(prefix + '.mdl', mdl)
  chainer.serializers.save_npz(prefix + '.opt', opt)