From e2cf6bbcc58cc769343a8c433288432e9572b535 Mon Sep 17 00:00:00 2001 From: Mike Clark Date: Tue, 7 Jul 2020 17:45:37 +0800 Subject: [PATCH] fix possible data leakage --- src/models/model.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/models/model.py b/src/models/model.py index 92cc0f8..feace22 100644 --- a/src/models/model.py +++ b/src/models/model.py @@ -136,7 +136,10 @@ class LatentModel(nn.Module): if target_y is not None: dist_post, log_var_post = self._latent_encoder(target_x, target_y) - z = dist_post.loc + if self.training: + z = dist_post.loc + else: + z = dist_prior.loc else: z = dist_prior.loc