search

rtqichen / torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
MIT License
5.61k stars 930 forks source link

RuntimeError: Mismatch in shape: grad_output[0] has a shape of torch.Size([1, 2404, 4]) and output[0] has a shape of torch.Size([1, 1963, 4]). #228

Open OswaldoBornemann opened 1 year ago

OswaldoBornemann commented 1 year ago

I have a 3D point cloud, which the number of each sample is vary. And i met the mismatch in shape problem during the back propagation, while i set the batch size to 2.

RuntimeError: Mismatch in shape: grad_output[0] has a shape of torch.Size([1, 2404, 4]) and output[0] has a shape of torch.Size([1, 1963, 4]).

I guess the problem came from torchdiffeq.