Skip to content

Commit

Permalink
Fix recommendation compression
Browse files Browse the repository at this point in the history
  • Loading branch information
Only-bottle committed Nov 29, 2024
1 parent 0864d2e commit 0d1c51a
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions netspresso/compressor/v2/compressor.py
Original file line number Diff line number Diff line change
Expand Up @@ -520,6 +520,8 @@ def recommendation_compression(
finally:
MetadataHandler.save_metadata(data=metadata, folder_path=output_dir)

return metadata

def automatic_compression(
self,
input_model_path: str,
Expand Down

0 comments on commit 0d1c51a

Please sign in to comment.