From: François Fleuret <francois@fleuret.org>
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)