Utilities

create_optimizers_cycle_gan

create_optimizers_cycle_gan(model: cycle_gan.model.CycleGAN, optim_cls: dict, **optim_params)

Creates optimizers (one for both generators and one per discriminator) holding the models’ parameters

Parameters:

• model (CycleGAN) – the model, whose parameters should be optimized
• optim_cls (dict) – dictionary containing the classes to create optimizers for the generator and the discriminator
• **optim_params – additional parameters to create the optimizers

Returns:

dictionary containing the different optimizers

Return type:

dict