From: François Fleuret Date: Wed, 18 Sep 2024 14:06:07 +0000 (+0200) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=ae616adc50d664dc1a02d926ac5d149d3cf15a51;p=culture.git Update. --- 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)