(
self,
ckpt_path: Union[None, str] = None,
ignore_keys: Union[Tuple, list] = (),
image_video_weights=[1, 1],
only_train_decoder=False,
context_parallel_size=0,
**kwargs,
)
| 500 | |
| 501 | class VideoAutoencodingEngine(AutoencodingEngine): |
| 502 | def __init__( |
| 503 | self, |
| 504 | ckpt_path: Union[None, str] = None, |
| 505 | ignore_keys: Union[Tuple, list] = (), |
| 506 | image_video_weights=[1, 1], |
| 507 | only_train_decoder=False, |
| 508 | context_parallel_size=0, |
| 509 | **kwargs, |
| 510 | ): |
| 511 | super().__init__(**kwargs) |
| 512 | self.context_parallel_size = context_parallel_size |
| 513 | if ckpt_path is not None: |
| 514 | self.init_from_ckpt(ckpt_path, ignore_keys=ignore_keys) |
| 515 | |
| 516 | def log_videos(self, batch: dict, additional_log_kwargs: Optional[Dict] = None, **kwargs) -> dict: |
| 517 | return self.log_images(batch, additional_log_kwargs, **kwargs) |
nothing calls this directly
no test coverage detected