Edit on GitHub
MMCV
DVCLive allows you to add experiment tracking capabilities to your OpenMMlab projects.
Usage
Register the
DvcliveLoggerHook
the following in the config file of your
OpenMMlab project:
log_config = dict(
interval=100,
hooks=[
dict(type='TextLoggerHook'),
dict(type='DvcliveLoggerHook')
]
)
Parameters
-
model_file
- (None
by default) - The name of the file where the model will be saved at the end of eachstep
. -
**kwargs
- Any additional arguments will be used to instantiate a newLive
instance.
Examples
- Using
**kwargs
to customizeLive
.
log_config = dict(
interval=100,
hooks=[
dict(type='TextLoggerHook'),
dict(
type='DvcliveLoggerHook',
dir="custom_dir"
)
]
)