Update.
authorFrançois Fleuret <francois@fleuret.org>
Sat, 22 Jun 2024 22:07:14 +0000 (00:07 +0200)
committerFrançois Fleuret <francois@fleuret.org>
Sat, 22 Jun 2024 22:07:14 +0000 (00:07 +0200)
main.py

diff --git a/main.py b/main.py
index 45fa68c..79b4b56 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -183,7 +183,7 @@ for n in vars(args):
 ######################################################################
 
 if args.check:
-    args.nb_train_samples = 500
+    args.nb_train_samples = 2500
     args.nb_test_samples = 100
 
 if args.physical_batch_size is None: