在 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"
)
]
)