Spaces:

OAOA
/

DifFace

Runtime error

DifFace / basicsr /models /video_gan_model.py

Zongsheng

first upload

06f26d7 over 1 year ago

No virus

507 Bytes

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