| from Architectures.ControllabilityGAN.wgan.init_weights import weights_init_D | |
| from Architectures.ControllabilityGAN.wgan.init_weights import weights_init_G | |
| from Architectures.ControllabilityGAN.wgan.resnet_1 import ResNet_D | |
| from Architectures.ControllabilityGAN.wgan.resnet_1 import ResNet_G | |
| def init_resnet(parameters): | |
| critic = ResNet_D(parameters['data_dim'][-1], parameters['size'], nfilter=parameters['nfilter'], nfilter_max=parameters['nfilter_max']) | |
| generator = ResNet_G(parameters['data_dim'][-1], parameters['z_dim'], parameters['size'], nfilter=parameters['nfilter'], | |
| nfilter_max=parameters['nfilter_max']) | |
| generator.apply(weights_init_G) | |
| critic.apply(weights_init_D) | |
| return generator, critic | |