From: François Fleuret Date: Wed, 18 Sep 2024 14:50:13 +0000 (+0200) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=6c1e5af6fdb00d97136df7ac6ae89bd51fb6ccf7;p=culture.git Update. --- diff --git a/main.py b/main.py index 464b217..a357687 100755 --- a/main.py +++ b/main.py @@ -483,6 +483,10 @@ def prioritized_rand(low): def ae_generate(model, nb, local_device=main_device): model.eval().to(local_device) + # We loop through the iterations first and through the + # mini-batches second so that we keep only the samples that have + # not stabilized + all_input = quiz_machine.pure_noise(nb, local_device) all_masks = all_input.new_full(all_input.size(), 1) all_changed = torch.full((all_input.size(0),), True, device=all_input.device)