Update.

diff --git a/main.py b/main.py
index 83227bb..f65bb8e 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -188,6 +188,7 @@ class TaskPicoCLVR(Task):
         self.device = device
         nb = args.data_size if args.data_size > 0 else 250000
 
+        log_string('generating {nb} samples (can take some time)')
         self.train_descr = generate_descr((nb * 4) // 5)
         self.test_descr = generate_descr((nb * 1) // 5)