File size: 480 Bytes
6be1ab7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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 SRGANModel:
init_training_settings
setup_optimizers
optimize_parameters
save
Then find functions in VideoBaseModel.
"""
|