basicsr/models/video_gan_model.py

from basicsr.utils.registry import MODEL_REGISTRY
from .srgan_model import SRGANModel
from .video_base_model import VideoBaseModel


@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`.

    """