在训练模型时,可以创建一个LossMonitor回调函数并传递给model.train()函数。该回调函数会在每个epoch结束后打印该epoch的平均损失值。同时,可以设置per_print_times参数来控制打印频率,例如每个batch打印一次或每个epoch打印一次。通过查看打印的信息,可以得到损失函数的曲线。

# 损失函数 net_loss = SoftmaxCrossEntropyWithLogitssparse=Truereduction=mean # sparse输出不是one hot编码时设为Ture # 优化器 lr = 0001 # 学习率 momentum = 09 # 动量 net_opt = nnMomentumnettrainable_params lr momentum #

原文地址: http://www.cveoy.top/t/topic/bgcG 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录