在 GitHub 上编辑
Live.log_params()
def log_params(params: Dict[ParamLike]):另请参阅 Live.log_param()。
用法
from dvclive import Live
live = Live()
params = {
"num_classes": 10,
"metrics": ["accuracy", "mae"],
"training": {
"epochs": 42,
"batch_size": 128,
"optimizer": {
"name": "adam",
"loss": "categorical_crossentropy",
"learning_rate": 1e-3,
},
},
}
live.log_params(params)描述
每次调用 Live.log_params(params) 时，DVCLive 会将 params 字典中的键值对写入 {Live.dir}/params.yaml：
lr: 1e-3
n_iter: 50可通过 dvc params 可视化已记录的参数：
dvc params diff dvclive/params.yaml若使用 DVC 流水线，参数依赖 会自动追踪，因此无需通过 DVCLive 手动记录它们。
参数
params- 包含待记录参数名称和值的字典。
异常
-
dvclive.error.InvalidParameterType- 当params字典中的任意参数值不属于支持的类型时抛出。支持的类型包括：ParamLike = Union[ int, float, str, bool, List["ParamLike"], Dict[str, "ParamLike"] ]