(self)
| 180 | return batch[self.input_key] |
| 181 | |
| 182 | def get_autoencoder_params(self) -> list: |
| 183 | params = [] |
| 184 | if hasattr(self.loss, "get_trainable_autoencoder_parameters"): |
| 185 | params += list(self.loss.get_trainable_autoencoder_parameters()) |
| 186 | if hasattr(self.regularization, "get_trainable_parameters"): |
| 187 | params += list(self.regularization.get_trainable_parameters()) |
| 188 | params = params + list(self.encoder.parameters()) |
| 189 | params = params + list(self.decoder.parameters()) |
| 190 | return params |
| 191 | |
| 192 | def get_discriminator_params(self) -> list: |
| 193 | if hasattr(self.loss, "get_trainable_parameters"): |
no test coverage detected