(self, z_var)
| 113 | return output_tensor |
| 114 | |
| 115 | def get_generator(self, z_var): |
| 116 | if cfg.GAN.NETWORK_TYPE == "default": |
| 117 | return self.generator(z_var) |
| 118 | else: |
| 119 | raise NotImplementedError |
| 120 | |
| 121 | # stage II generator (hr_g) |
| 122 | def residual_block(self, x_c_code): |