flytekitplugins.onnxpytorch.PyTorch2ONNX¶

class flytekitplugins.onnxpytorch.PyTorch2ONNX(model: 'Union[torch.nn.Module, torch.jit.ScriptModule, torch.jit.ScriptFunction]' = None)[source]¶

Methods

Parameters:: model (Module | ScriptModule | ScriptFunction)

classmethod from_dict(kvs, *, infer_missing=False)¶

Parameters:: kvs (dict | list | str | int | float | bool | None)
Return type:: A

classmethod from_json(s, *, parse_float=None, parse_int=None, parse_constant=None, infer_missing=False, **kw)¶

Parameters:: s (str | bytes | bytearray)
Return type:: A

classmethod schema(*, infer_missing=False, only=None, exclude=(), many=False, context=None, load_only=(), dump_only=(), partial=False, unknown=None)¶

Parameters:

infer_missing (bool)
many (bool)
partial (bool)

Return type:

SchemaF[A]

to_dict(encode_json=False)¶

Return type:: Dict[str, dict | list | str | int | float | bool | None]

to_json(*, skipkeys=False, ensure_ascii=True, check_circular=True, allow_nan=True, indent=None, separators=None, default=None, sort_keys=False, **kw)¶

Parameters:

skipkeys (bool)
ensure_ascii (bool)
check_circular (bool)
allow_nan (bool)
indent (int | str | None)
separators (Tuple[str, str] | None)
default (Callable | None)
sort_keys (bool)

Return type:

str

Attributes

dataclass_json_config = None

model: Module | ScriptModule | ScriptFunction = None