开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

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

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

展开
收起
cuicuicuic 2023-10-19 12:29:37 39 0
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
    赞同 展开评论 打赏

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

相关电子书

更多
Python第四讲——使用IPython/Jupyter Notebook与日志服务玩转超大规模数据分析与可视化 立即下载
大规模日志数据平台架构面临的问题与挑战 立即下载
PostgresChina2018_赖思超_PostgreSQL10_hash索引的WAL日志修改版final 立即下载