在 GitHub 上编辑
MMCV
DVCLive 可以为您的 OpenMMlab 项目添加实验跟踪功能。
用法
在您的 OpenMMlab 项目的配置文件中注册 DvcliveLoggerHook:
log_config = dict(
interval=100,
hooks=[
dict(type='TextLoggerHook'),
dict(type='DvcliveLoggerHook')
]
)参数
-
model_file-(默认为None)- 每个step结束时模型将被保存的文件名。 -
**kwargs- 任何其他参数都将用于实例化一个新的Live实例。
示例
- 使用
**kwargs自定义Live。
log_config = dict(
interval=100,
hooks=[
dict(type='TextLoggerHook'),
dict(
type='DvcliveLoggerHook',
dir="custom_dir"
)
]
)