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