search

neuralmagic / sparsezoo

Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes
Apache License 2.0
371 stars 25 forks source link

Print model analysis #423

Closed horheynm closed 10 months ago

horheynm commented 10 months ago

Description

Added print at the end of cli runs and __repr__

(.venv) george@gpuserver6:~/sparsezoo$ sparsezoo.analyze resnet_v1-101-imagenet-base
Downloading (…)se/model.onnx.tar.gz: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 159M/159M [00:06<00:00, 26.4MB/s]
Params:
        total           : 44442816
        sparsity%       : 0.0
        size [bits]     : 1422170112.0
        quantized %     : 0.0
Ops:
        total           : 15320195048
        sparsity%       : 0.0
        size [bits]     : 489428172032.0
        quantized %     : 0.0
Memory Access:
        total           : 4071724290048
        sparsity%       : 0.0021713690245701222
        size [bits]     : 130295177281536.0
        quantized %     : 0.0
rahul-tuli commented 10 months ago

Should we change the base branch to #421 so we can cherry pick this in one go?