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

[Fix] val loader should not drop last by default. (#857)

Browse Source
This commit is contained in:
Fei Wang 2022-09-28 08:22:23 +08:00 committed by GitHub
parent 4d73607fb8
commit 0143e5fdb0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mmcls/apis/train.py
View File

@ -209,6 +209,7 @@ def train_model(model,
**loader_cfg, **loader_cfg,
'shuffle': False, # Not shuffle by default 'shuffle': False, # Not shuffle by default
'sampler_cfg': None, # Not use sampler by default 'sampler_cfg': None, # Not use sampler by default
'drop_last': False, # Not drop last by default
**cfg.data.get('val_dataloader', {}), **cfg.data.get('val_dataloader', {}),
} }
val_dataloader = build_dataloader(val_dataset, **val_loader_cfg) val_dataloader = build_dataloader(val_dataset, **val_loader_cfg)