← Python API
vraven · class

CaptureConfig

CaptureConfig(preferred_routes: 'tuple[CaptureRoute, ...]' = (<CaptureRoute.EXPORT: 'torch_export'>, <CaptureRoute.FX: 'torch_fx'>, <CaptureRoute.DISPATCH: 'runtime_dispatch'>, <CaptureRoute.MODULE_HOOKS: 'module_hooks'>), strict_export: 'bool' = True, functionalise_export: 'bool' = True, dynamic_shapes: 'Any | None' = None, auto_dynamic_batch: 'bool' = True, validate_equivalence: 'bool' = True, validate_gradients: 'bool' = True, absolute_tolerance: 'float' = 1e-05, relative_tolerance: 'float' = 0.0001, dispatch_include_view_ops: 'bool' = True, max_stack_trace_chars: 'int' = 4000) -> None

Controls capture route order and validation behaviour.

Usage

from vraven import CaptureConfig

value = CaptureConfig(...)
Source of truthThis page is generated from the checked-out public symbol, signature, and docstring. Pin the package version for reproducible work.