Keeping channels as 3

2023-04-12 17:29:50 +02:00 · 2023-04-12 17:29:50 +02:00 · a83149f61e
commit a83149f61e
parent 1f4667a08d
1 changed files with 0 additions and 1 deletions
--- a/DPI/models.py
+++ b/DPI/models.py
@ -88,7 +88,6 @@ class ObservationDecoder(nn.Module):
        out = self.dense(features)
        out = torch.reshape(out, [-1, self.input_size, 1, 1])
        out = self.convtranspose(out)
-        
        mean = torch.reshape(out, (*out_batch_shape, *self.output_shape))
        out_dist = torch.distributions.independent.Independent(torch.distributions.Normal(mean, 1), len(self.output_shape))
        return out_dist