From 0143e5fdb0f7f8b0e7be932f64408e2074eb82f7 Mon Sep 17 00:00:00 2001 From: Fei Wang Date: Wed, 28 Sep 2022 08:22:23 +0800 Subject: [PATCH] [Fix] val loader should not drop last by default. (#857) --- mmcls/apis/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mmcls/apis/train.py b/mmcls/apis/train.py index ed3a37b9b..40e9531d4 100644 --- a/mmcls/apis/train.py +++ b/mmcls/apis/train.py @@ -209,6 +209,7 @@ def train_model(model, **loader_cfg, 'shuffle': False, # Not shuffle by default 'sampler_cfg': None, # Not use sampler by default + 'drop_last': False, # Not drop last by default **cfg.data.get('val_dataloader', {}), } val_dataloader = build_dataloader(val_dataset, **val_loader_cfg)