BatchCLIPSeg error

When I try using BatchCLIPSeg I get error:

```
Error occurred when executing BatchCLIPSeg:

Input and output must have the same number of spatial dimensions, but got input with spatial dimensions of [1, 352, 352] and output size of (832, 1216). Please provide input tensor in (N, C, d1, d2, ...,dK) format and output size in (o1, o2, ...,oK) format.
```

Any idea how to fix?