From 9c65231a3a3628c97a86dc9cf16a323b646288dd Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Sun, 30 Sep 2018 16:43:51 +0800
Subject: [PATCH] save mmdet version in checkpoint as meta info
---
.gitignore | 1 +
tools/train.py | 4 +++-
2 files changed, 4 insertions(+), 1 deletion(-)
diff --git a/.gitignore b/.gitignore
index d5ef5f5..01c47d6 100644
--- a/.gitignore
+++ b/.gitignore
@@ -105,4 +105,5 @@ venv.bak/
# cython generated cpp
mmdet/ops/nms/*.cpp
+mmdet/version.py
data
diff --git a/tools/train.py b/tools/train.py
index dde6b06..78f8a55 100644
--- a/tools/train.py
+++ b/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)
--
GitLab