(self, batch: AttrDict, options: Optional[AttrDict] = None)
| 55 | self.latent_warp = latent_warp_from_config(latent_warp, device=device) |
| 56 | |
| 57 | def forward(self, batch: AttrDict, options: Optional[AttrDict] = None) -> AttrDict: |
| 58 | h = self.encode_to_bottleneck(batch, options=options) |
| 59 | return self.bottleneck_to_params(h, options=options) |
| 60 | |
| 61 | def encode_to_bottleneck( |
| 62 | self, batch: AttrDict, options: Optional[AttrDict] = None |
nothing calls this directly
no test coverage detected