Weights uploading error

[Ehteshamciitwah]

Hello, Thank you for your excellent work. I am trying to run test.py but model uploading is showing an error.
Pretrained provided weights contain an extra variable (part_emb_3_level) but build model doesn't have such a variable. Furthermore, Can you please provide the exact model as described on paper. PartNet.py computing some variable but not using them such supp_sim_loss_level_2 query_sim_loss_level_2 weighted_part_map_level_2 ori_part_map_level_2 Any solution?

[wu-zhonghua]

Sorry, we initialize ''part_emb_3_level'' but not use in the forward function. You can add ''self.part_emb_3_level = nn.Parameter(torch.FloatTensor(1, 256, self.part_num).normal_(0.0, variance))'' in the init() function