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