fix possible data leakage

2026-06-27 18:03:39 +08:00 · 2020-07-05 09:02:34 +08:00
parent 9252601f49
commit f7e21ec6ef
1 changed files with 1 additions and 1 deletions
@@ -324,7 +324,7 @@ class NeuralProcess(nn.Module):
            
        dist_prior, log_var_prior = self._latent_encoder(context_x, context_y)

-        if target_y is not None:
+        if (target_y is not None) and self.training:
            target_y2 = self.norm_y(target_y)
            if self._use_rnn:
                target_y2, _ = self._lstm_y(target_y2)