ModelScope我看代码里有metric但是为啥日志只保存了loss?

ModelScope我看代码里有metric但是为啥日志只保存了loss?92649e145565681508e96925910f5d36.png

展开
收起
真的很搞笑 2023-10-19 12:29:37 115 分享 版权
1 条回答
写回答
取消 提交回答
  • 可能是因为在训练过程中,metric没有被正确地计算和记录。你可以尝试以下方法来解决这个问题:

    1. 确保你的模型定义中包含了计算metric的代码。例如,如果你使用的是PyTorch,可以在forward函数中添加计算metric的代码。

    2. 在训练循环中,确保将metric的值添加到日志中。例如,如果你使用的是torch.utils.tensorboard,可以这样做:

    from torch.utils.tensorboard import SummaryWriter
    
    writer = SummaryWriter()
    
    for epoch in range(num_epochs):
        # ...训练过程...
    
        # 计算metric
        metric_value = compute_metric(output, target)
    
        # 将metric值添加到日志中
        writer.add_scalar('Metric/YourMetricName', metric_value, epoch)
    
    1. 在训练结束后,使用writer.close()关闭日志文件。
    2023-10-19 13:47:24
    赞同 展开评论

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

热门讨论

热门文章

还有其他疑问?
咨询AI助理