From: François Fleuret <francois@fleuret.org>
Date: Sun, 18 Feb 2024 22:30:01 +0000 (+0100)
Subject: Update.
X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=picoclvr.git;a=commitdiff_plain;h=4f489998d6e73680c3a031e8932a7678c16268e3

Update.
---

diff --git a/tasks.py b/tasks.py
index 78910a0..1ea3b5d 100755
--- a/tasks.py
+++ b/tasks.py
@@ -144,6 +144,9 @@ class TaskFromFile(Task):
         )
         self.test_input, self.test_pred_masks = self.tensorize(pairs[nb_train_samples:])
 
+        assert self.train_input.size(0) == nb_train_samples
+        assert self.test_input.size(0) == nb_test_samples
+
     def batches(self, split="train", nb_to_use=-1, desc=None):
         assert split in {"train", "test"}
         input = self.train_input if split == "train" else self.test_input