Github
BasicSR
latest
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
StyleGAN2Model
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: latest
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds