Github
BasicSR
stable
API
basicsr API
basicsr.archs
basicsr.data
basicsr.losses
basicsr.metrics
basicsr.models
basicsr.models.__init__
basicsr.models.base_model
basicsr.models.edvr_model
basicsr.models.esrgan_model
basicsr.models.hifacegan_model
basicsr.models.lr_scheduler
basicsr.models.realesrgan_model
basicsr.models.realesrnet_model
basicsr.models.sr_model
basicsr.models.srgan_model
basicsr.models.stylegan2_model
basicsr.models.swinir_model
basicsr.models.video_base_model
basicsr.models.video_gan_model
basicsr.models.video_recurrent_gan_model
basicsr.models.video_recurrent_model
basicsr.ops
basicsr.test
basicsr.train
basicsr.utils
scripts API
BasicSR
»
basicsr API
»
basicsr.models.stylegan2_model
Edit on GitHub
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]
Read the Docs
v: stable
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds