save mmdet version in checkpoint as meta info

.gitignore

@@ -105,4 +105,5 @@ venv.bak/
 
 # cython generated cpp
 mmdet/ops/nms/*.cpp
+mmdet/version.py
 data

tools/train.py

@@ -9,7 +9,7 @@ import torch
 from mmcv import Config
 from mmcv.torchpack import Runner, obj_from_dict
 
-from mmdet import datasets
+from mmdet import datasets, __version__
 from mmdet.core import (init_dist, DistOptimizerHook, DistSamplerSeedHook,
                         MMDataParallel, MMDistributedDataParallel,
                         CocoDistEvalRecallHook, CocoDistEvalmAPHook)
@@ -89,6 +89,8 @@ def main():
     if args.work_dir is not None:
         cfg.work_dir = args.work_dir
     cfg.gpus = args.gpus
+    # add mmdet version to checkpoint as meta data
+    cfg.checkpoint_config.meta = dict(mmdet_version=__version__)
 
     logger = get_logger(cfg.log_level)