Почему выходные параметры tvm.relay.optimize пусты?
Я новичок в использовании TVM. Я отследил файл resnet50. Теперь я хочу получить оптимизированные модули, используя: model = ResNet50(10, 3) model.to(dtype)
trace_input = torch.rand(1, 3, 224, 224)
traced_model = torch.jit.trace(model, trace_input)
for p in traced_model.parameters():
p.requires_grad_(False)
shape_list = [(i.debugName().split('.')[0], i.type().sizes())
for i in list(traced_model.graph.inputs())[1:]]
mod, mod_params = tvm.relay.frontend.from_pytorch(traced_model, shape_list, default_dtype=dtype_str)
mod_optimized, mod_params_optimized = tvm.relay.optimize(mod, target, params=mod_params)
Однако длина словаря mode_params_optimized равна нулю. Я был бы признателен, если бы кто-нибудь мог предложить мне, как найти проблему.