Почему выходные параметры 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 равна нулю. Я был бы признателен, если бы кто-нибудь мог предложить мне, как найти проблему.

0 ответов

Другие вопросы по тегам