(self, z_var)
| 141 | return output_tensor |
| 142 | |
| 143 | def get_generator(self, z_var): |
| 144 | if cfg.GAN.NETWORK_TYPE == "default": |
| 145 | return self.generator(z_var) |
| 146 | elif cfg.GAN.NETWORK_TYPE == "simple": |
| 147 | return self.generator_simple(z_var) |
| 148 | else: |
| 149 | raise NotImplementedError |
| 150 | |
| 151 | # d-net |
| 152 | def context_embedding(self): |
no test coverage detected