From ae616adc50d664dc1a02d926ac5d149d3cf15a51 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Wed, 18 Sep 2024 16:06:07 +0200 Subject: [PATCH] Update. --- main.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/main.py b/main.py index d8dffe2..c6eedfb 100755 --- a/main.py +++ b/main.py @@ -380,8 +380,12 @@ def add_noise(imt_set): return torch.cat([input[:, None], masks[:, None], targets[:, None]], dim=1) +###################################################################### + # IMT for input / masks / target +# Generate a batch for prediction + def batch_for_prediction_imt(input): nb = input.size(0) -- 2.39.5