From 33b317f1ce0b1f955f86dbdcce6275818882e0bf Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Tue, 3 Sep 2024 09:04:59 +0200 Subject: [PATCH] Update. --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 9b2282f..fb0b4df 100755 --- a/main.py +++ b/main.py @@ -1138,7 +1138,7 @@ def targets_and_prediction(model, input, mask_generate, prompt_noise=0.0): if prompt_noise > 0: mask_prompt_noise = ( - torch.rand(input.size(), device=input.device()) <= prompt_noise + torch.rand(input.size(), device=input.device) <= prompt_noise ).long() noise = torch.randint( quiz_machine.problem.nb_colors, input.size(), device=input.device -- 2.39.5