Merge remote-tracking branch 'origin/master' into dev

This commit is contained in:
mzr1996 2022-01-13 19:27:23 +08:00
commit 321ad09e6d

View File

@ -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,