使用MindSpore训练及保存模型

MindSpore提供了回调Callback机制,可以在训练过程中执行自定义逻辑,这里以使用框架提供的ModelCheckpoint为例。

  1. ModelCheckpoint可以保存网络模型和参数,以便进行后续的Fine-tuning(微调)操作。

    from mindspore.train.callback import ModelCheckpoint, CheckpointConfig

    # 设置模型保存参数

    config_ck = CheckpointConfig(save_checkpoint_steps=1875, keep_checkpoint_max=10)

    # 应用模型保存参数

    ckpoint = ModelCheckpoint(prefix="checkpoint_lenet", config=config_ck)

  2. 通过MindSpore提供的model.train接口可以方便地进行网络的训练,LossMonitor可以监控训练过程中loss值的变化。

    # 导入模型训练需要的库

    from mindspore.nn import Accuracy

    from mindspore.train.callback import LossMonitor

    from mindspore import Model

  3. def train_net(args, model, epoch_size, data_path, repeat_size, ckpoint_cb, sink_mode):

        """定义训练的方法"""

        # 加载训练数据集

        ds_train = create_dataset(os.path.join(data_path, "train"), 32, rep

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值