Update.

[mygpt.git] / 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 ])