Closed paran3 closed 3 months ago
develop
from netspresso import NetsPresso
EMAIL = "YOUR_EMAIL" PASSWORD = "YOUR_PASSWORD"
netspresso = NetsPresso(email=EMAIL, password=PASSWORD)
compressor = netspresso.compressor()
INPUT_SHAPES = [{"batch": 1, "channel": 3, "dimension": [224, 224]}] INPUT_MODEL_PATH = "./examples/sample_models/graphmodule.pt" OUTPUT_DIR = "./outputs/compressed/pytorch_automatic_compression_1" COMPRESSION_RATIO = 0.5
compression_result = compressor.automatic_compression( input_shapes=INPUT_SHAPES, input_model_path=INPUT_MODEL_PATH, output_dir=OUTPUT_DIR, compression_ratio=COMPRESSION_RATIO, )
No response
Latest release (v1.7.0)
We have decided to resolve this issue by removing the v1 module entirely instead of making any modifications.
Describe the bug
Description
Checklists
develop
) branch.Have you searched existing issues? 🔎
Reproduction
from netspresso import NetsPresso
EMAIL = "YOUR_EMAIL" PASSWORD = "YOUR_PASSWORD"
netspresso = NetsPresso(email=EMAIL, password=PASSWORD)
1. Declare compressor
compressor = netspresso.compressor()
2. Set variables for compression
INPUT_SHAPES = [{"batch": 1, "channel": 3, "dimension": [224, 224]}] INPUT_MODEL_PATH = "./examples/sample_models/graphmodule.pt" OUTPUT_DIR = "./outputs/compressed/pytorch_automatic_compression_1" COMPRESSION_RATIO = 0.5
3. Run automatic compression
compression_result = compressor.automatic_compression( input_shapes=INPUT_SHAPES, input_model_path=INPUT_MODEL_PATH, output_dir=OUTPUT_DIR, compression_ratio=COMPRESSION_RATIO, )
Screenshot
No response
Logs
No response
System Info