(self, mu, logstd)
| 64 | self.output_logits = output_logits |
| 65 | |
| 66 | def reparameterize(self, mu, logstd): |
| 67 | std = torch.exp(logstd) |
| 68 | eps = torch.randn_like(std) |
| 69 | return mu + eps * std |
| 70 | |
| 71 | def forward_conv(self, obs,flatten=True): |
| 72 | if obs.max() > 1.: |
nothing calls this directly
no outgoing calls
no test coverage detected