Update.
authorFrancois Fleuret <francois@fleuret.org>
Sat, 16 Jul 2022 10:12:06 +0000 (12:12 +0200)
committerFrancois Fleuret <francois@fleuret.org>
Sat, 16 Jul 2022 10:12:06 +0000 (12:12 +0200)
main.py

diff --git a/main.py b/main.py
index a18beb1..83c305b 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -131,8 +131,8 @@ class TaskPicoCLVR(Task):
             many_colors = many_colors
         )
 
-        self.test_descr = descr[:nb // 5]
-        self.train_descr = descr[nb // 5:]
+        self.test_descr = descr[:nb // 5]
+        self.train_descr = descr[nb // 5:]
 
         descr = [ s.strip().split(' ') for s in descr ]
         l = max([ len(s) for s in descr ])