tb_writer.add_scalar(tags[0], mean_loss, epoch) tb_writer.add_scalar(tags[1], acc, epoch) tb_writer.add_scalar(tags[2], optimizer.param_groups[0]["lr"], epoch)
时间: 2023-05-18 09:03:19 浏览: 225
open_loop.rar_SCALAR CONTROL_control_pmsm
这是一段 PyTorch 中的代码,用于在 TensorBoard 中记录训练过程中的损失、准确率和学习率等信息。其中,tb_writer 是一个 TensorBoard 的写入器,tags 是一个包含三个字符串的列表,分别表示损失、准确率和学习率的标签,mean_loss、acc 和 optimizer.param_groups[0]["lr"] 分别表示平均损失、准确率和当前学习率。
阅读全文