MindSpore自定义在train的过程中实时验证的回调函数

扶我起，还能学

于 2021-07-12 15:58:50 发布

阅读量1.2k

点赞数 2

CC 4.0 BY-SA版权

文章标签： python 深度学习

本文链接：https://blog.csdn.net/yang_zm/article/details/118676858

本文介绍了如何在MindSpore中实现自定义回调函数进行训练过程中的实时验证。通过继承`Callback`类并定义`EvalCallBack`，设置每隔一定epoch数执行验证，并输出模型的Accuracy指标。在训练过程中，`epoch_end`方法被调用时，如果满足验证条件，便进行验证并打印准确率。这有助于监控模型的训练效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本文讲述MindSpore自定义在train的过程中实时验证的回调函数，继承callback类自定义evalcallback，可以用来设置：每隔几个epoch进行验证，实时输出model指定的metrics评价指标。然后通过实例化对象，将这个回调过程放入model.train()中的callback()中。

简单来讲就是输出对训练集的实时model.eval()测试结果。

class EvalCallBack(Callback):
    def __init__(self, model, eval_dataset, epochs_to_eval, per_eval, dataset_sink_mode):
        self.model = mod

最低0.47元/天解锁文章

200万优质内容无限畅学