X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=f65bb8eafde0e667d23f0d12746a480383bf85ff;hb=f082ce9255f87b6c1bfebfac00f94820a16e04f1;hp=83227bb43a24897b506f0b82035a69dc33f7acfe;hpb=82ddf9ca322e6fcc8f9364a696c26d15841d13d8;p=mygpt.git

diff --git a/main.py b/main.py
index 83227bb..f65bb8e 100755
--- 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)