Compiling MaskRCNN from Torchvision model zoo issues

abdulazizm · December 20, 2021, 5:30am

Compiling MaskRCNN from Torchvision model zoo issues

opened 08:31AM - 16 Dec 21 UTC

JIT traced maskrcnn model from model zoo and provided desired layouts but couldn…'t avoid Attribute error. Any workarounds for this? ``` mod, params = relay.frontend.from_pytorch(script_module, shape_list) mod = relay.transform.InferType()(mod) mod["main"] = bind_params_by_name(mod["main"], params) mod = transform.RemoveUnusedFunctions()(mod) # no warnings when included these in desired layouts #desired_layouts = {'nn.conv2d': ['NHWC', 'default'], 'image.resize2d': ['NHWC'], 'vision.roi_align': ['NHWC','default'], 'nn.conv2d_transpose':['NHWC','default']} desired_layouts = {'nn.conv2d': ['NHWC', 'default']} seq = tvm.transform.Sequential([relay.transform.RemoveUnusedFunctions(), relay.transform.ConvertLayout(desired_layouts), relay.transform.FoldConstant()]) with tvm.transform.PassContext(opt_level=3): mod = seq(mod) mod = relay.transform.DynamicToStatic()(mod) # getting below mentioned here even after using DynamicToStatic() mod = partition_for_vitis_ai(mod, params, dpu=target) print(mod['main']) ``` `AttributeError: <class 'tvm.tir.expr.Any'> has no attribute value` @jornt-xilinx Any suggestion on this?

@masahi Any suggestions on this?