search

stevenygd / PointFlow

PointFlow : 3D Point Cloud Generation with Continuous Normalizing Flows
https://www.guandaoyang.com/PointFlow/
MIT License
720 stars 101 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]). #38

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.