在 GitHub 上编辑

Live()

初始化一个 DVCLive 日志记录器。

class Live:

    def __init__(
        self,
        dir: str = "dvclive",
        resume: bool = False,
        report: Literal["md", "notebook", "html", None] = None,
        save_dvc_exp: bool = True,
        dvcyaml: Optional[str] = "dvc.yaml",
        cache_images: bool = False,
        exp_name: Optional[str] = None,
        exp_message: Optional[str] = None,
        monitor_system: bool = False,
    ):

用法

from dvclive import Live

with Live() as live:
    ...

描述

要记录机器学习参数、指标和其他元数据,必须使用 Live() 实例。

除非 resume=True,否则 Live() 会删除 dir 下所有与 DVCLive 相关的文件。

你可以将 Live() 用作上下文管理器。退出上下文管理器时,将自动调用 Live.end()

属性

  • step - 请参见 Live.next_step()

  • summary - 请参见 Live.make_summary()

  • dir - 存储 输出 的目录路径。

  • dvc_file - dvc.yaml 文件的路径。

  • metrics_file - {Live.dir}/metrics.json

  • params_file - {Live.dir}/params.yaml

  • plots_dir - {Live.dir}/plots

  • report_file - {Live.dir}/report.{format}。其中 format 可为 HTML(.html)或 Markdown(.md),具体取决于 report 参数的值。

参数

  • dir - 保存 DVCLive 输出的位置,默认为 dvclive

  • resume - 若为 True,DVCLive 将尝试从 metrics_file 中读取上一次的 step 并从该点继续。默认为 False

  • report - 可选值为 htmlnotebookmdNone。详见 Live.make_report()。默认为 None

  • save_dvc_exp - 若为 True,DVCLive 将在 Live.end() 执行时创建一个新的 DVC 实验。默认为 True

    若在 DVC 流水线 中使用 DVCLive 并通过 dvc exp run 运行,则此选项将被忽略。

  • dvcyaml - 指定 dvc.yaml 文件的写入位置,该文件会在 Live.next_step()Live.end() 中添加用于指标、图表和参数的 DVC 配置。若为 None,则不写入 dvc.yaml 文件。默认为 dvc.yaml。详见 Live.make_dvcyaml()

    若传入类似 subdir/dvc.yaml 的字符串,DVCLive 将把配置写入该路径(文件必须命名为 dvc.yaml)。

    若为 False,DVCLive 将不写入 dvc.yaml(适用于你独立跟踪 DVCLive 的指标、图表和参数,以避免重复)。

  • cache_images - 若为 True,DVCLive 将在 Live.end() 中缓存通过 Live.log_image() 记录的所有图像。默认为 False

    若运行 DVC 流水线cache_images 将被忽略,应将图像作为流水线的 输出 进行缓存。

  • exp_name - 若不为 Nonesave_dvc_expTrue,则提供的字符串将作为参数传递给 dvc exp save --name

    如果在 dvc exp run 内部使用 DVCLive,该选项将被忽略,请改用 dvc exp run --name

  • exp_message — 如果不为 Nonesave_dvc_expTrue,则提供的字符串将传递给 dvc exp save --message

    如果在 dvc exp run 内部使用 DVCLive,该选项将被忽略,请改用 dvc exp run --message

  • monitor_system — 如果为 True,DVCLive 将记录 系统指标,包括 GPU、CPU、RAM 和磁盘使用情况。默认值为 False

方法

内容

🐛 发现问题?告诉我们!或者修复它:

在 GitHub 上编辑

有疑问?加入我们的聊天,我们会为您提供帮助:

Discord 聊天