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).
