From 57d8f6716cff814dc520472967d64a0c8a8fc58d Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Thu, 5 Sep 2024 14:48:44 +0200 Subject: [PATCH] Update. --- main.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/main.py b/main.py index 8562e32..174b9b8 100755 --- a/main.py +++ b/main.py @@ -1429,6 +1429,9 @@ for n_epoch in range(current_epoch, args.nb_epochs): threads = [] + # for model in models: + # log_string(f"DEBUG {model.id} {sum([ p.sum() for p in model.parameters()]).item()}") + start_time = time.perf_counter() for gpu, model in zip(gpus, weakest_models): -- 2.39.5