diff --git a/tools/train.py b/tools/train.py
index c9049a495..2880f8c77 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -158,13 +158,15 @@ def main():
         val_dataset = copy.deepcopy(cfg.data.val)
         val_dataset.pipeline = cfg.data.train.pipeline
         datasets.append(build_dataset(val_dataset))
-    if cfg.checkpoint_config is not None:
-        # save mmcls version, config file content and class names in
-        # checkpoints as meta data
-        cfg.checkpoint_config.meta = dict(
+
+    # save mmcls version, config file content and class names in
+    # runner as meta data
+    meta.update(
+        dict(
             mmcls_version=__version__,
             config=cfg.pretty_text,
-            CLASSES=datasets[0].CLASSES)
+            CLASSES=datasets[0].CLASSES))
+
     # add an attribute for visualization convenience
     train_model(
         model,