basicsr.models.stylegan2_model

class basicsr.models.stylegan2_model.StyleGAN2Model(opt)[source]

Bases: BaseModel

StyleGAN2 model.

dist_validation(dataloader, current_iter, tb_logger, save_img)[source]
feed_data(data)[source]
init_training_settings()[source]
make_noise(batch, num_noise)[source]
mixing_noise(batch, prob)[source]
nondist_validation(dataloader, current_iter, tb_logger, save_img)[source]
optimize_parameters(current_iter)[source]
save(epoch, current_iter)[source]

Save networks and training state.

setup_optimizers()[source]
test()[source]