File size: 507 Bytes
a64b7d4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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`.
"""
|