When I set batch_size to 2, the output of the encoder is abnormal, as shown below:
This will lead to problems in the final result, how to solve it?
B, N, _, _ = input.shape # type: int, int, int, int
ValueError: too many values to unpack (expected 4)
When I set batch_size to 2, the output of the encoder is abnormal, as shown below:
This will lead to problems in the final result, how to solve it?
ValueError: too many values to unpack (expected 4)