fastvideo.fastvideo_args
#
The arguments of FastVideo Inference.
Module Contents#
Classes#
Enumeration for different pipeline modes. |
|
Training arguments. Inherits from FastVideoArgs and adds training-specific arguments. If there are any conflicts, the training arguments will take precedence. |
|
Enumeration for different workload types. |
Functions#
Prepare the inference arguments from the command line arguments. |
|
Temporarily set the current fastvideo config. Used during model initialization. We save the current fastvideo config in a global variable, so that all modules can access it, e.g. custom ops can access the fastvideo config to determine how to dispatch. |
Data#
API#
- class fastvideo.fastvideo_args.ExecutionMode[source]#
-
Enumeration for different pipeline modes.
Inherits from str to allow string comparison for backward compatibility.
Initialization
Initialize self. See help(type(self)) for accurate signature.
- classmethod from_string(value: str) fastvideo.fastvideo_args.ExecutionMode [source]#
Convert string to ExecutionMode enum.
- class fastvideo.fastvideo_args.FastVideoArgs[source]#
- STA_mode: fastvideo.configs.pipelines.base.STA_Mode[source]#
None
- static add_cli_args(parser: fastvideo.utils.FlexibleArgumentParser) fastvideo.utils.FlexibleArgumentParser [source]#
- classmethod from_cli_args(args: argparse.Namespace) fastvideo.fastvideo_args.FastVideoArgs [source]#
- classmethod from_kwargs(**kwargs: Any) fastvideo.fastvideo_args.FastVideoArgs [source]#
- pipeline_config: fastvideo.configs.pipelines.base.PipelineConfig[source]#
‘field(…)’
- preprocess_config: fastvideo.configs.configs.PreprocessConfig | None[source]#
None
- workload_type: fastvideo.fastvideo_args.WorkloadType[source]#
None
- class fastvideo.fastvideo_args.TrainingArgs[source]#
Bases:
fastvideo.fastvideo_args.FastVideoArgs
Training arguments. Inherits from FastVideoArgs and adds training-specific arguments. If there are any conflicts, the training arguments will take precedence.
- static add_cli_args(parser: fastvideo.utils.FlexibleArgumentParser) fastvideo.utils.FlexibleArgumentParser [source]#
- classmethod from_cli_args(args: argparse.Namespace) fastvideo.fastvideo_args.TrainingArgs [source]#
- class fastvideo.fastvideo_args.WorkloadType[source]#
-
Enumeration for different workload types.
Inherits from str to allow string comparison for backward compatibility.
Initialization
Initialize self. See help(type(self)) for accurate signature.
- classmethod from_string(value: str) fastvideo.fastvideo_args.WorkloadType [source]#
Convert string to WorkloadType enum.
- fastvideo.fastvideo_args.get_current_fastvideo_args() fastvideo.fastvideo_args.FastVideoArgs [source]#
- fastvideo.fastvideo_args.prepare_fastvideo_args(argv: list[str]) fastvideo.fastvideo_args.FastVideoArgs [source]#
Prepare the inference arguments from the command line arguments.
- Parameters:
argv – The command line arguments. Typically, it should be
sys.argv[1:]
to ensure compatibility withparse_args
when no arguments are passed.- Returns:
The inference arguments.
- fastvideo.fastvideo_args.set_current_fastvideo_args(fastvideo_args: fastvideo.fastvideo_args.FastVideoArgs)[source]#
Temporarily set the current fastvideo config. Used during model initialization. We save the current fastvideo config in a global variable, so that all modules can access it, e.g. custom ops can access the fastvideo config to determine how to dispatch.