From d41399de6a68131c936f0b310d2c07da9d0aa425 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Tue, 3 Sep 2024 15:22:14 +0200 Subject: [PATCH] Update. --- main.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 2376868..6113813 100755 --- a/main.py +++ b/main.py @@ -1606,7 +1606,10 @@ for n_epoch in range(current_epoch, args.nb_epochs): save_badness_statistics(last_n_epoch_c_quizzes, models, c_quizzes, "after") last_n_epoch_c_quizzes = n_epoch - nb_c_quizzes_to_generate = args.nb_train_samples // args.c_quiz_multiplier + nb_gpus = len(gpus) + nb_c_quizzes_to_generate = ( + args.nb_train_samples // args.c_quiz_multiplier + nb_gpus - 1 + ) // nb_gpus # -------------------------------------------------------------------- -- 2.39.5