blob: cc4447690a47562b3a95dd937e40b419335b07a3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
from r_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.
"""
|
