add artifact logging to wandb hook

[fcakyon]

Problem: Currently it is not possible to upload experiment config files and model weights to wandb. However it very crucial to properly track experiments.

Solution: With this PR, users can upload the contents in work_dir to wandb after training ends utilizing log_artifact() method of wandb logger: https://docs.wandb.ai/ref/python/run#log_artifact

Usage:

log_config = dict(
    interval=1,
    hooks=[
        dict(
            type="WandbLoggerHook",
            init_kwargs=dict(
                entity=WANDB_ENTITY,
                project=WANDB_PROJECT,
                name=WANDB_NAME,
            ),
            log_artifact=True,
            out_suffix=('.log.json', '.log', '.py', '.pth')
        ),
    ],
)

[zhouzaida]

Hi @fcakyon, Should we provide an option to upload only the specified file suffix like mmcv/runner/hooks/logger/text.py , since some files are too large and not always needed.

https://docs.wandb.ai/ref/python/artifact#add_file

[fcakyon]

Hi @fcakyon, Should we provide an option to upload only the specified file suffix like mmcv/runner/hooks/logger/text.py , since some files are too large and not always needed.

https://docs.wandb.ai/ref/python/artifact#add_file

@zhouzaida its done!

[zhouzaida]

Hi @fcakyon, Should we provide an option to upload only the specified file suffix like mmcv/runner/hooks/logger/text.py , since some files are too large and not always needed.
https://docs.wandb.ai/ref/python/artifact#add_file

@zhouzaida its done!

👍

[zhouzaida]

It would be great if some unit tests are added.

[fcakyon]

It would be great if some unit tests are added.

its also done 👍

[zhouzaida]

It would be great if some unit tests are added.

its also done 👍

Wow, it is so great.

[fcakyon]

@zhouzaida it is ready 👍

[teamwong111]

It seems that the docstring lacks an args with_step.

[fcakyon]

@teamwong111 i have no idea what with_step parameter does, i did not add that parameter, it was there before.

[zhouzaida]

@teamwong111 i have no idea what with_step parameter does, i did not add that parameter, it was there before.

The details about the argument can be found at #913

[fcakyon]

@zhouzaida it is ready to be merged now 👍

[teamwong111]

Nice, thank you for your contribution

[fcakyon]

@zhouzaida @teamwong111 when will this be merged? i need it urgently :)

[zhouzaida]

@zhouzaida @teamwong111 when will this be merged? i need it urgently :)

Hi @fcakyon , we will merge the PR in the next few days.