摘要:
前面几篇分别是数据导入模块,模型类的构建模块,损失函数模块,优化器模块,上述模块搭建好后建立训练模块,将上述几个模块在训练模块中调用完成训练。 这个不太好解释直接看模块示例: 上面是训练模块,分别传入训练总轮次steps,实例化模型model,损失函数loss_func,但是下面训练的时候它将损失函 阅读全文
posted @ 2024-09-02 22:03
千主
阅读(36)
评论(0)
推荐(0)
2024年9月2日