Skip to main content

create_gan_optimizer

trustformers_optim::task_specific

Function create_gan_optimizer 

Source 
pub fn create_gan_optimizer(
    g_lr: f32,
    d_lr: f32,
    use_spectral_norm: bool,
) -> GANOptimizer