yolov10 ReduceMax keepdims == 0

[sihaiwuhu]

Hi, there is a panic like this when building from yolov10.onnx model. How should I solve this error?

ERROR burn_import::logger: PANIC => panicked at /home/ubuntu/.cargo/registry/src/index.crates.io-1949cf8c6b5b557f/burn-import-0.16.0/src/onnx/op_configuration.rs:1467:9:
ReduceMax: the reduce operation must preserve the reduced dimension

--- stderr
thread 'main' panicked at /home/ubuntu/.cargo/registry/src/index.crates.io-1949cf8c6b5b557f/burn-import-0.16.0/src/onnx/op_configuration.rs:1467:9:
ReduceMax: the reduce operation must preserve the reduced dimension
note: run with RUST_BACKTRACE=1 environment variable to display a backtrace
warning: build failed, waiting for other jobs to finish...
ubuntu@PC-PW09NVP9:/mnt/d/workspace/rust_ws/burn_demo1$

[laggui]

That's because keepdims is set to false, and that's not currently supported when converting the model to Burn code.

burn/crates/burn-import/src/onnx/op_configuration.rs

Lines 1465 to 1468 in dd628ec

	if !axes.is_empty() && keepdims == 0 {
	// Not supported in Burn
	panic!("ReduceMax: the reduce operation must preserve the reduced dimension")
	}

In burn, tensor.max_dim(dim) returns a tensor of the same rank (with a dim of size 1 where it was reduced). The support could be expanded in the codegen to chain a max operation with .squeeze(dim) when keepdims is false.