From dca0fad05259b45658600220ebf93a4663e8191e Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sat, 24 Aug 2024 16:16:46 +0200 Subject: [PATCH] Update. --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index e4bb494..e13c148 100755 --- a/main.py +++ b/main.py @@ -947,7 +947,7 @@ def test_ae(local_device=main_device): model.to(local_device).train() optimizer_to(model.optimizer, local_device) - nb_iterations = 5 + nb_iterations = 10 for n_epoch in range(args.nb_epochs): # ---------------------- -- 2.39.5