Source code for basicsr.models.video_gan_model
from basicsr.utils.registry import MODEL_REGISTRY
from .srgan_model import SRGANModel
from .video_base_model import VideoBaseModel
[docs]@MODEL_REGISTRY.register()
class VideoGANModel(SRGANModel, VideoBaseModel):
"""Video GAN model.
Use multiple inheritance.
It will first use the functions of :class:`SRGANModel`:
- :func:`init_training_settings`
- :func:`setup_optimizers`
- :func:`optimize_parameters`
- :func:`save`
Then find functions in :class:`VideoBaseModel`.
"""