【问题标题】:Logging training metrics to a csv file将训练指标记录到 csv 文件
【发布时间】:2021-06-23 06:25:52
【问题描述】:

我想在使用 pytorch 编写的YOLOV5 上进行训练时将所有训练指标记录到 csv 文件中,但问题是我不想使用 tensorboard。

为了实现这个目标,我尝试了如下一些技术:

-先登录到tensorboard,然后尝试转成csv文件(失败)

-从权重和偏差中提取日志文件(失败)

-训练时直接写入csv文件(失败)

这里我尝试打开一个 csv 文件并将张量数据转换为字符串,但没有成功。

所以我真的很想知道我怎样才能实现这个目标,因为 YOLOV5 没有使用像 model.fit 这样我们可以使用回调的 pytorch 实用程序。

提前致谢。

【问题讨论】:

    标签: pytorch yolov5


    【解决方案1】:

    您可能会尝试使用权重和偏见yolov5集成。 这是链接:https://docs.wandb.ai/guides/integrations/yolov5

    该链接有更多细节,但这里有一些引号传达了基本思想:

    只是通过安装WandB,您将激活内置的W&B日志记录功能:系统指标,模型指标和媒体记录到交互式仪表板。

    pip install wandb
    git clone https://github.com/ultralytics/yolov5.git
    python yolov5/train.py  # train a small network on a small dataset
    

    只是按照Wandb.pip安装Wandb git克隆

    的链接遵循打印到标准的链接

    您还可以进行模型版本控制和数据集可视化,这在W&B Yolov5集成文档页面(上面链接)中会更详细地解释。您还可以为指南观看此YouTube视频:https://youtu.be/yyecuhBmLxE

    【讨论】:

    • 虽然此链接可能会回答问题,但最好在此处包含答案的基本部分并提供链接以供参考。如果链接页面发生更改,仅链接的答案可能会失效。
    • 谢谢@ tyler2p,我添加了一些细节。 span>
    猜你喜欢
    • 2018-09-28
    • 2020-11-01
    • 2021-05-08
    • 2022-01-22
    • 1970-01-01
    • 2021-11-16
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多