def select(config):
""" select different subtask
"""
if config.target == 'kinvae.encoder':
return KINVAE_ENCODER(config)
elif config.target == 'kinvae.encoder2':
return KINVAE_ENCODER2(config)
elif config.target == 'kinvae.encoder3':
return KINVAE_ENCODER3(config)
elif config.target == 'kinvae.encoder4':
return KINVAE_ENCODER4(config)
elif config.target == 'kinvae.bidirect':
return KINVAE_BIDIRECT(config)
elif config.target == 'kinvae.bidirect2':
return KINVAE_BIDIRECT2(config)
elif config.target == 'kinvae.bidirect3':
return KINVAE_BIDIRECT3(config)
elif config.target == 'kinvae.bidirect4':
return KINVAE_BIDIRECT4(config)
elif config.target == 'kinvae.bidirect5':
return KINVAE_BIDIRECT5(config)
elif config.target == 'kinvae.bidirect6':
return KINVAE_BIDIRECT6(config)
elif config.target == 'kinvae.bidirect7':
return KINVAE_BIDIRECT7(config)
elif config.target == 'kinvae.bidirect8':
return KINVAE_BIDIRECT8(config)
elif config.target == 'kinvae.bidirect9':
return KINVAE_BIDIRECT9(config)
elif config.target == 'kinvae.bidirect10':
return KINVAE_BIDIRECT10(config)
elif config.target == 'kinvae.bidirect11':
return KINVAE_BIDIRECT11(config)
elif config.target == 'kinvae.feature':
return KINVAE_FEATURE(config)
else:
raise ValueError('Unknown Target [%s]' % config.target)
def __init__(self, config):
KINVAE_BIDIRECT.__init__(self, config)
self.is_save_all_images = False
self.is_save_batch_images = True
self.is_save_feats = True
def __init__(self, config): KINVAE_BIDIRECT.__init__(self, config)