CenterNet icon indicating copy to clipboard operation
CenterNet copied to clipboard

when i generate onnx file ,i got the error.

Open LCWdmlearning opened this issue 5 years ago • 1 comments

config["system"]["snapshot_name"] = crf system_config = SystemConfig().update_config(config["system"]) model_file = "core.models.{}".format(crf) model_file = importlib.import_module(model_file) model_c = model_file.model(db) x = torch.randn(1, 3, 255, 255, requires_grad=True) torch_out = torch.onnx._export(model_c, x, "super_resolution.onnx", export_params=True)

error : RuntimeError: ONNX export failed: Couldn't export Python operator TopPoolFunction

Can you give me some suggestion?

LCWdmlearning avatar May 13 '19 09:05 LCWdmlearning

@LCWdmlearning Hi, I meet a same error, Did you solve it?

TTMRonald avatar Jun 04 '19 08:06 TTMRonald