Skip to content
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 each step.

  • **kwargs - Any additional arguments will be used to instantiate a new Live instance.

Examples

  • Using model_file.
log_config = dict(
    interval=100,
    hooks=[
        dict(type='TextLoggerHook'),
        dict(type='DvcliveLoggerHook', model_file="my_model.pth")
    ]
)
  • Using **kwargs to customize Live.
log_config = dict(
    interval=100,
    hooks=[
        dict(type='TextLoggerHook'),
        dict(
            type='DvcliveLoggerHook',
            model_file="my_model.pth",
            dir="custom_dir"
        )
    ]
)
Content

🐛 Found an issue? Let us know! Or fix it:

Edit on GitHub

Have a question? Join our chat, we will help you:

Discord Chat