[Apache TVM Discuss] [Questions] How could I extract prams(weight) information from static mod(tvm.IRModule) in python

[Chi Fang via Apache TVM Discuss]

the following is an example about how I get mod:

    md                = "mobilenet_v1_1.0_224_original.onnx"
    dict    = {input.0: [1,3,224,224]}
    onnx_model = onnx.load(model_path)
    mod, params = relay.frontend.from_onnx(onnx_model, dict, 
freeze_params=True) # for unknown shape


now params is empy {}
how could I get params(weight) list and its shape for calculating total weight 
size to decide if my device has enough memory to handle this mod.

any suggestion?

thanks~~~





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/how-could-i-extract-prams-weight-information-from-static-mod-tvm-irmodule-in-python/10419/1)
 to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.apache.org/email/unsubscribe/895863e41fbe68d172c886b8cb7e94e83dd65b800110c95cd9fa157b0ea358a8).