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- (Noneby 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 newLiveinstance.
Examples
- Using
**kwargsto customizeLive.
log_config = dict(
interval=100,
hooks=[
dict(type='TextLoggerHook'),
dict(
type='DvcliveLoggerHook',
dir="custom_dir"
)
]
)