Skip to content

Commit

Permalink
Dont throw an error when we can't figure out the parameter count (#210)
Browse files Browse the repository at this point in the history
* Dont throw an error when we can't figure out the parameter count

* rev version number
  • Loading branch information
jeremyfowers authored Aug 7, 2024
1 parent abbf3cb commit 81a3b74
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 5 deletions.
5 changes: 1 addition & 4 deletions src/turnkeyml/common/analyze_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,12 +27,9 @@ def count_parameters(model: torch.nn.Module) -> int:
if tensor.name not in onnx_model.graph.input
)
)
elif isinstance(model, str) and model.endswith(".yaml"):
else:
return None

# Raise exception if an unsupported model type is provided
raise AnalysisException(f"model type {type(model)} is not supported")


def get_onnx_ops_list(onnx_model) -> Dict:
"""
Expand Down
2 changes: 1 addition & 1 deletion src/turnkeyml/version.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__ = "3.0.3"
__version__ = "3.0.4"

0 comments on commit 81a3b74

Please sign in to comment.