CenterNet
CenterNet copied to clipboard
when i generate onnx file ,i got the error.
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 Hi, I meet a same error, Did you solve it?