| 280 | |
| 281 | |
| 282 | class EncodingWithNetwork(nn.Module): |
| 283 | def __init__(self, encoding, network): |
| 284 | super().__init__() |
| 285 | self.encoding, self.network = encoding, network |
| 286 | |
| 287 | def forward(self, x): |
| 288 | return self.network(self.encoding(x)) |
| 289 | |
| 290 | def update_step(self, epoch, global_step): |
| 291 | update_module_step(self.encoding, epoch, global_step) |
| 292 | update_module_step(self.network, epoch, global_step) |
| 293 | |
| 294 | |
| 295 | def get_encoding_with_network(n_input_dims, n_output_dims, encoding_config, network_config): |
no outgoing calls
no test coverage detected