def from_options(options, discriminator):
if options.loss in Registry.keys('loss'):
cls = Registry.get('loss', options.loss)
return cls(options, discriminator)
else:
raise RegistryError(
f'missing value {options.loss} for namespace `loss`')
def from_options(options, generator, discriminator, loss, hooks):
if options.game in Registry.keys('game'):
cls = Registry.get('game', options.game)
return cls(options, generator, discriminator, loss, hooks)
else:
raise RegistryError(
f'missing value {options.game} for namespace `game`')
def from_options(options):
if options.log in Registry.keys('log'):
cls = Registry.get('log', options.log)
return cls(options)
else:
raise RegistryError(
f'missing value {options.log} for namespace `log`')
def from_options(options):
if options.data_format in Registry.keys('data'):
cls = Registry.get('data', options.data_format)
return cls(options)
else:
raise RegistryError(
f'missing value {options.data_format} for namespace `data`')
def from_options(options):
if options.noise in Registry.keys('noise'):
cls = Registry.get('noise', options.noise)
return cls(options)
else:
raise RegistryError(
f'missing value {options.noise} for namespace `noise`')
def from_options(options):
if options.evaluation_criterion in Registry.keys('evaluation'):
cls = Registry.get('evaluation', options.evaluation_criterion)
return cls(options)
else:
raise RegistryError(
f'missing value {options.evaluation_criterion} for namespace `evaluation`'
)
def from_options(options):
if options.save_images_as in Registry.keys('snapshot'):
cls = Registry.get('snapshot', options.save_images_as)
return cls(options)
else:
raise RegistryError(
f'missing value {options.save_images_as} for namespace `snapshot`'
)
def from_options(options):
if options.generator in Registry.keys('generator'):
cls = Registry.get('generator', options.generator)
generator = cls(options)
else:
raise RegistryError(
f'missing value {options.generator} for namespace `generator`')
if options.restore:
state_dict = torch.load(
os.path.join(options.model_dir, options.experiment,
'generator.pt'))
generator.load_state_dict(state_dict)
generator = generator.to(options.device)
if options.parallel:
generator = nn.DataParallel(generator)
return generator