crtie / leveraging-se-3-equivariance-for-learning-3d-geometric-shape-assembly Goto Github PK

Hi, thanks for sharing your code.

I'm running the script on breaking bad dataset and getting this error. Any idea how to resolve this?

By the way, the set_trace in line 56 in the same file should be comment out, otherwise the code will break in that line. Is this the final version of your code?

Dear authors,

I had tried to run the train.py, and I find that the 'recon_pts' and 'whole_recon_pcs' are not included in 'pre_dict'. Is there any way to get these two values?

Hi, thank you for sharing the code.

The bug occurs when I edit the config file to use the VN_Regressor to predict the pose. So I wonder if the program supports the VN_Regressor.
Here's where the bug occurs:
File "Leveraging-SE-3-Equivariance-for-Learning-3D-Geometric-Shape-Assembly/BreakingBad/multi
_part_assembly/models/vnn/network.py", line 217, in forward
R_6d, trans = self._predict_pose(
File "Leveraging-SE-3-Equivariance-for-Learning-3D-Geometric-Shape-Assembly/BreakingBad/multi
_part_assembly/models/vnn/network.py", line 167, in _predict_pose
valid_R_6d, valid_trans = self.pose_predictor(
...
File "Leveraging-SE-3-Equivariance-for-Learning-3D-Geometric-Shape-Assembly/BreakingBad/multi
_part_assembly/models/vnn/modules.py", line 152, in forward
f = self.fc_layers(x)
...
File "Leveraging-SE-3-Equivariance-for-Learning-3D-Geometric-Shape-Assembly/BreakingBad/multi
_part_assembly/models/vnn/vn_layers.py", line 33, in forward
x_out = self.map_to_feat(x.transpose(1,-1)).transpose(1,-1)
File "/root/miniconda3/envs/assemblySE3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _ca
ll_impl
return forward_call(*input, **kwargs)
File "/root/miniconda3/envs/assemblySE3/lib/python3.8/site-packages/torch/nn/modules/linear.py", line 103, in forw
ard
return F.linear(input, self.weight, self.bias)
File "/root/miniconda3/envs/assemblySE3/lib/python3.8/site-packages/torch/nn/functional.py", line 1848, in linear
return torch._C._nn.linear(input, weight, bias)
RuntimeError: mat1 and mat2 shapes cannot be multiplied (93x1024 and 512x1024)

Hi again, variable quat_A in line 227 in file 'GSM/shape_assembly/datasets/baseline/dataloader_CR.py' is not define. Any idea how to resolve this?

Dear author,

When I run 'python scripts/train.py --cfg_file configs/vnn/vnn-everyday.py', it shows 'No such file or directory ../dataverse_files/everyday.train.txt', it seems like there is a txt file got lost, could you please tell me how to get it?