X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;ds=sidebyside;f=main.py;h=b01ea0ae078f0e4a494b25f2971d64c2b358bfe0;hb=c3621f9a75cd4d79410d90a29dc9fdec401eaa2d;hp=83227bb43a24897b506f0b82035a69dc33f7acfe;hpb=52c6bd98650c846459f10e8303dd2e6c7ba2a68f;p=mygpt.git

diff --git a/main.py b/main.py
index 83227bb..b01ea0a 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(f'generating {nb} samples (can take some time)')
         self.train_descr = generate_descr((nb * 4) // 5)
         self.test_descr = generate_descr((nb * 1) // 5)