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