From c967a2ac88e075082473f3ca219660fa12cf67d1 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 23 Mar 2021 10:26:29 +0800 Subject: Fix indexing bugs in validation dataset selector --- models/model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'models/model.py') diff --git a/models/model.py b/models/model.py index 766e513..17716ad 100644 --- a/models/model.py +++ b/models/model.py @@ -152,7 +152,8 @@ class Model: train_size = dataset_config.get('train_size', 74) val_dataset_config['train_size'] = train_size + val_size val_dataset_config['selector']['classes'] = ClipClasses({ - str(c).zfill(3) for c in range(train_size, train_size + val_size) + str(c).zfill(3) + for c in range(train_size + 1, train_size + val_size + 1) }) val_dataset = self._parse_dataset_config(val_dataset_config) val_dataloader = iter(self._parse_dataloader_config( -- cgit v1.2.3