From c967a2ac88e075082473f3ca219660fa12cf67d1 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
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(-)

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