X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=958dfc70a791d2b47708b10e7f44e53e7456fc2e;hb=545b06257dd69285c6321d7bd713d819da74953b;hp=9198edc25ce5501faef8e5dd8b74d65e9a22836e;hpb=1eef58fd084437bbcd2041b946b468615e203dd8;p=picoclvr.git diff --git a/main.py b/main.py index 9198edc..958dfc7 100755 --- a/main.py +++ b/main.py @@ -413,6 +413,7 @@ if args.task == "file": nb_train_samples=args.nb_train_samples, nb_test_samples=args.nb_test_samples, batch_size=args.batch_size, + shuffle=True, device=device, ) args.max_percents_of_test_in_train = 0