mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
Merge remote-tracking branch 'origin/master' into dev
This commit is contained in:
commit
321ad09e6d
@ -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,
|
||||
|
Loading…
x
Reference in New Issue
Block a user