Problem loading the pretrained model in PCN

[himmetozcan]

Hi there. Thanks for this repository. I think there are some problems in model constructions, the model files and pretrained model dictionary seem to be not matching. I see problems while loading a pretrained model. Such as in point completion network:

from learning3d.models import PCN
import torch
pcn = PCN()
pcn.load_state_dict(torch.load('learning3d/pretrained/exp_pcn/models/best_model.t7', map_location='cpu'))

and this is the error I get:

RuntimeError: Error(s) in loading state_dict for PCN:
        Missing key(s) in state_dict: "conv5.weight", "conv5.bias", "conv6.weight", "conv6.bias", "conv7.weight", "conv7.bias". 

Do you know what the problem is in here?