mirrors/mmclassification
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 21:53:55 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
mzr1996 2022-01-13 19:27:23 +08:00
commit 321ad09e6d
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tools/train.py
View File

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