Hi, According to the previous topic, [image] Mask_rcnn pruned model inference error TAO Toolkit Hi. I get this error when inferencing the mask_rcnn pruned model: File "/root/.cache/bazel/_bazel_root/ed34e6d125608f91724fda23656f1726/execroot/ai_infra/bazel-out/k8-fastbui…

Need not care about the tlt model’s size, the tensorrt engine will be smaller than the unpruned one.

Mask RCNN pruning problem

Accelerated Computing Intelligent Video Analytics TAO Toolkit

p.vahidinia December 31, 2021, 7:14am 4

Hi,
Yes, you are right. By pruning, the size of the model decreases. But according to this link, the mask rcnn pruned model must be retrained first and so, the model size increases again.