文件记录器

如果你希望在训练开始时，指定一些文件复制到日志目录（run开头的目录下），那么非常推荐你使用LogdirFileWriter插件。

改进插件

SwanLab插件均为开源代码，你可以在Github源代码中查看，欢迎提交你的建议和PR！

插件用法

1. 初始化LogdirFileWriter：

from swanlab.plugin.writer import LogdirFileWriter

logdirfile_writer = LogdirFileWriter(
    sub_dir="code",
    files=[
        "config.yaml",
        "README.md",
    ]
)

• sub_dir参数如果不为None，则在run目录下创建1个sub_dir文件夹来保存文件
• files参数指定了需要复制的文件列表（也支持仅传入1个str）。

2. 传入插件：

swanlab.init(
    ...
    callbacks=[logdirfile_writer]
)

执行代码后，就会在logdir下对应的run开头目录下将files参数中的文件复制到该目录中（如果设置了sub_dir参数，则会复制到该子目录下）。

示例代码

from swanlab.plugin.writer import LogdirFileWriter
import swanlab

logdirfile_writer = LogdirFileWriter(
    sub_dir="code",
    file_path=["package.json", "README.md"],
)

swanlab.init(project="test-plugin", callbacks=[logdirfile_writer])

swanlab.log({"loss": 0.2, "acc": 0.9})
swanlab.finish()

外部注册插件

如果你使用的是SwanLab与其他框架的集成，故而不太好找到swanlab.init，那么你可以使用swanlab.register_callbacks方法，在外部传入插件：

import swanlab

# 等价于 swanlab.init(callbaks=[...])
swanlab.register_callbacks([...])