Spaces:
Running
on
Zero
Running
on
Zero
File size: 752 Bytes
6faeba1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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
|