We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8bd88bb commit 87946cfCopy full SHA for 87946cf
src/lightly_train/_commands/export_task.py
@@ -75,7 +75,7 @@ def export_onnx(
75
num_channels: int = 3,
76
height: int = 224,
77
width: int = 224,
78
- precision: str = Literal["32-true", "16-true"],
+ precision: Literal["32-true", "16-true"] = "32-true",
79
verify: bool = True,
80
overwrite: bool = False,
81
format_args: dict[str, Any] | None = None,
@@ -92,7 +92,7 @@ def _export_task(
92
93
94
95
96
97
98
0 commit comments