Cannot Load autoencoder_25_squares & Poor Performance on Pre-trained Model

[HanXuan-Li]

Hello, thanks a lot for sharing.
I want to use the pre-trained autoencoder model to meshing point cloud data(2048 points sampling from ShapeNet). I encountered two problems:

Poor performance on pre-trained autoencoder_1_sphere Model

When I used python train.py --demo --demo_input_path my_pcd.ply --reload_model_path ./training/trained_models/atlasnet_autoencoder_1_sphere/network.pth to generate mesh on my own data, I got this:

My data is just like the examples shown in your paper:

I also tried to use your data(2ca6df7a5377825cfee773c7de26c274.points.ply) but still got a poor result like this:

So I'd like to know if I did something wrong.

Cannot Load autoencoder_25_squares

I used python train.py --demo --demo_input_path my_pcd.ply --reload_model_path ./training/trained_models/atlasnet_autoencoder_25_squares/network.pth to load the autoencoder_25_squares model, but got this:

Loaded compiled 3D CUDA chamfer distance
Launching new HTTP instance in port 8891
TMUX=0 tmux new-session -d -s http_server \; send-keys "/home/lihanxuan/anaconda3/envs/atlasnet/bin/python -m http.server -p 8891 > /dev/null 2>&1" Enter
duplicate session: http_server
Setting up a new session...
/home/lihanxuan/anaconda3/envs/atlasnet/lib/python3.6/site-packages/torchvision/transforms/transforms.py:258: UserWarning: Argument interpolation should be of type InterpolationMode instead of int. Please, use InterpolationMode enum.
  "Argument interpolation should be of type InterpolationMode instead of int. "
New MLP decoder : hidden size 512, num_layers 2, activation relu
Network weights loaded from  ./training/trained_models/atlasnet_autoencoder_25_squares/network.pth!
Traceback (most recent call last):
  File "train.py", line 20, in <module>
    trainer.build_network()
  File "/home/lihanxuan/Project/_AtlasNet/model/trainer_model.py", line 31, in build_network
    self.reload_network()
  File "/home/lihanxuan/Project/_AtlasNet/model/trainer_model.py", line 42, in reload_network
    self.network.module.load_state_dict(torch.load(self.opt.reload_model_path, map_location='cuda:0'))
  File "/home/lihanxuan/anaconda3/envs/atlasnet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1224, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for EncoderDecoder:
        Unexpected key(s) in state_dict: "decoder.decoder.1.conv1.weight", "decoder.decoder.1.conv1.bias", "decoder.decoder.1.conv2.weight", "decoder.decoder.1.conv2.bias", "decoder.decoder.1.conv_list.0.weight", "decoder.decoder.1.conv_list.0.bias", "decoder.decoder.1.conv_list.1.weight", "decoder.decoder.1.conv_list.1.bias", "decoder.decoder.1.last_conv.weight", "decoder.decoder.1.last_conv.bias", "decoder.decoder.1.bn1.weight", "decoder.decoder.1.bn1.bias", "decoder.decoder.1.bn1.running_mean", "decoder.decoder.1.bn1.running_var", "decoder.decoder.1.bn1.num_batches_tracked", "decoder.decoder.1.bn2.weight", "decoder.decoder.1.bn2.bias", "decoder.decoder.1.bn2.running_mean", "decoder.decoder.1.bn2.running_var", "decoder.decoder.1.bn2.num_batches_tracked", "decoder.decoder.1.bn_list.0.weight", "decoder.decoder.1.bn_list.0.bias", "decoder.decoder.1.bn_list.0.running_mean", "decoder.decoder.1.bn_list.0.running_var", "decoder.decoder.1.bn_list.0.num_batches_tracked", "decoder.decoder.1.bn_list.1.weight", "decoder.decoder.1.bn_list.1.bias", "decoder.decoder.1.bn_list.1.running_mean", "decoder.decoder.1.bn_list.1.running_var", "decoder.decoder.1.bn_list.1.num_batches_tracked", "decoder.decoder.2.conv1.weight", "decoder.decoder.2.conv1.bias", "decoder.decoder.2.conv2.weight", "decoder.decoder.2.conv2.bias", "decoder.decoder.2.conv_list.0.weight", "decoder.decoder.2.conv_list.0.bias", "decoder.decoder.2.conv_list.1.weight", "decoder.decoder.2.conv_list.1.bias", "decoder.decoder.2.last_conv.weight", "decoder.decoder.2.last_conv.bias", "decoder.decoder.2.bn1.weight", "decoder.decoder.2.bn1.bias", "decoder.decoder.2.bn1.running_mean", "decoder.decoder.2.bn1.running_var", "decoder.decoder.2.bn1.num_batches_tracked", "decoder.decoder.2.bn2.weight", "decoder.decoder.2.bn2.bias", "decoder.decoder.2.bn2.running_mean", "decoder.decoder.2.bn2.running_var", "decoder.decoder.2.bn2.num_batches_tracked", "decoder.decoder.2.bn_list.0.weight", "decoder.decoder.2.bn_list.0.bias", "decoder.decoder.2.bn_list.0.running_mean", "decoder.decoder.2.bn_list.0.running_var", "decoder.decoder.2.bn_list.0.num_batches_tracked", "decoder.decoder.2.bn_list.1.weight", "decoder.decoder.2.bn_list.1.bias", "decoder.decoder.2.bn_list.1.running_mean", "decoder.decoder.2.bn_list.1.running_var", "decoder.decoder.2.bn_list.1.num_batches_tracked", "decoder.decoder.3.conv1.weight", "decoder.decoder.3.conv1.bias", "decoder.decoder.3.conv2.weight", "decoder.decoder.3.conv2.bias", "decoder.decoder.3.conv_list.0.weight", "decoder.decoder.3.conv_list.0.bias", "decoder.decoder.3.conv_list.1.weight", "decoder.decoder.3.conv_list.1.bias", "decoder.decoder.3.last_conv.weight", "decoder.decoder.3.last_conv.bias", "decoder.decoder.3.bn1.weight", "decoder.decoder.3.bn1.bias", "decoder.decoder.3.bn1.running_mean", "decoder.decoder.3.bn1.running_var", "decoder.decoder.3.bn1.num_batches_tracked", "decoder.decoder.3.bn2.weight", "decoder.decoder.3.bn2.bias", "decoder.decoder.3.bn2.running_mean", "decoder.decoder.3.bn2.running_var", "decoder.decoder.3.bn2.num_batches_tracked", "decoder.decoder.3.bn_list.0.weight", "decoder.decoder.3.bn_list.0.bias", "decoder.decoder.3.bn_list.0.running_mean", "decoder.decoder.3.bn_list.0.running_var", "decoder.decoder.3.bn_list.0.num_batches_tracked", "decoder.decoder.3.bn_list.1.weight", "decoder.decoder.3.bn_list.1.bias", "decoder.decoder.3.bn_list.1.running_mean", "decoder.decoder.3.bn_list.1.running_var", "decoder.decoder.3.bn_list.1.num_batches_tracked", "decoder.decoder.4.conv1.weight", "decoder.decoder.4.conv1.bias", "decoder.decoder.4.conv2.weight", "decoder.decoder.4.conv2.bias", "decoder.decoder.4.conv_list.0.weight", "decoder.decoder.4.conv_list.0.bias", "decoder.decoder.4.conv_list.1.weight", "decoder.decoder.4.conv_list.1.bias", "decoder.decoder.4.last_conv.weight", "decoder.decoder.4.last_conv.bias", "decoder.decoder.4.bn1.weight", "decoder.decoder.4.bn1.bias", "decoder.decoder.4.bn1.running_mean", "decoder.decoder.4.bn1.running_var", "decoder.decoder.4.bn1.num_batches_tracked", "decoder.decoder.4.bn2.weight", "decoder.decoder.4.bn2.bias", "decoder.decoder.4.bn2.running_mean", "decoder.decoder.4.bn2.running_var", "decoder.decoder.4.bn2.num_batches_tracked", "decoder.decoder.4.bn_list.0.weight", "decoder.decoder.4.bn_list.0.bias", "decoder.decoder.4.bn_list.0.running_mean", "decoder.decoder.4.bn_list.0.running_var", "decoder.decoder.4.bn_list.0.num_batches_tracked", "decoder.decoder.4.bn_list.1.weight", "decoder.decoder.4.bn_list.1.bias", "decoder.decoder.4.bn_list.1.running_mean", "decoder.decoder.4.bn_list.1.running_var", "decoder.decoder.4.bn_list.1.num_batches_tracked", "decoder.decoder.5.conv1.weight", "decoder.decoder.5.conv1.bias", "decoder.decoder.5.conv2.weight", "decoder.decoder.5.conv2.bias", "decoder.decoder.5.conv_list.0.weight", "decoder.decoder.5.conv_list.0.bias", "decoder.decoder.5.conv_list.1.weight", "decoder.decoder.5.conv_list.1.bias", "decoder.decoder.5.last_conv.weight", "decoder.decoder.5.last_conv.bias", "decoder.decoder.5.bn1.weight", "decoder.decoder.5.bn1.bias", "decoder.decoder.5.bn1.running_mean", "decoder.decoder.5.bn1.running_var", "decoder.decoder.5.bn1.num_batches_tracked", "decoder.decoder.5.bn2.weight", "decoder.decoder.5.bn2.bias", "decoder.decoder.5.bn2.running_mean", "decoder.decoder.5.bn2.running_var", "decoder.decoder.5.bn2.num_batches_tracked", "decoder.decoder.5.bn_list.0.weight", "decoder.decoder.5.bn_list.0.bias", "decoder.decoder.5.bn_list.0.running_mean", "decoder.decoder.5.bn_list.0.running_var", "decoder.decoder.5.bn_list.0.num_batches_tracked", "decoder.decoder.5.bn_list.1.weight", "decoder.decoder.5.bn_list.1.bias", "decoder.decoder.5.bn_list.1.running_mean", "decoder.decoder.5.bn_list.1.running_var", "decoder.decoder.5.bn_list.1.num_batches_tracked", "decoder.decoder.6.conv1.weight", "decoder.decoder.6.conv1.bias", "decoder.decoder.6.conv2.weight", "decoder.decoder.6.conv2.bias", "decoder.decoder.6.conv_list.0.weight", "decoder.decoder.6.conv_list.0.bias", "decoder.decoder.6.conv_list.1.weight", "decoder.decoder.6.conv_list.1.bias", "decoder.decoder.6.last_conv.weight", "decoder.decoder.6.last_conv.bias", "decoder.decoder.6.bn1.weight", "decoder.decoder.6.bn1.bias", "decoder.decoder.6.bn1.running_mean", "decoder.decoder.6.bn1.running_var", "decoder.decoder.6.bn1.num_batches_tracked", "decoder.decoder.6.bn2.weight", "decoder.decoder.6.bn2.bias", "decoder.decoder.6.bn2.running_mean", "decoder.decoder.6.bn2.running_var", "decoder.decoder.6.bn2.num_batches_tracked", "decoder.decoder.6.bn_list.0.weight", "decoder.decoder.6.bn_list.0.bias", "decoder.decoder.6.bn_list.0.running_mean", "decoder.decoder.6.bn_list.0.running_var", "decoder.decoder.6.bn_list.0.num_batches_tracked", "decoder.decoder.6.bn_list.1.weight", "decoder.decoder.6.bn_list.1.bias", "decoder.decoder.6.bn_list.1.running_mean", "decoder.decoder.6.bn_list.1.running_var", "decoder.decoder.6.bn_list.1.num_batches_tracked", "decoder.decoder.7.conv1.weight", "decoder.decoder.7.conv1.bias", "decoder.decoder.7.conv2.weight", "decoder.decoder.7.conv2.bias", "decoder.decoder.7.conv_list.0.weight", "decoder.decoder.7.conv_list.0.bias", "decoder.decoder.7.conv_list.1.weight", "decoder.decoder.7.conv_list.1.bias", "decoder.decoder.7.last_conv.weight", "decoder.decoder.7.last_conv.bias", "decoder.decoder.7.bn1.weight", "decoder.decoder.7.bn1.bias", "decoder.decoder.7.bn1.running_mean", "decoder.decoder.7.bn1.running_var", "decoder.decoder.7.bn1.num_batches_tracked", "decoder.decoder.7.bn2.weight", "decoder.decoder.7.bn2.bias", "decoder.decoder.7.bn2.running_mean", "decoder.decoder.7.bn2.running_var", "decoder.decoder.7.bn2.num_batches_tracked", "decoder.decoder.7.bn_list.0.weight", "decoder.decoder.7.bn_list.0.bias", "decoder.decoder.7.bn_list.0.running_mean", "decoder.decoder.7.bn_list.0.running_var", "decoder.decoder.7.bn_list.0.num_batches_tracked", "decoder.decoder.7.bn_list.1.weight", "decoder.decoder.7.bn_list.1.bias", "decoder.decoder.7.bn_list.1.running_mean", "decoder.decoder.7.bn_list.1.running_var", "decoder.decoder.7.bn_list.1.num_batches_tracked", "decoder.decoder.8.conv1.weight", "decoder.decoder.8.conv1.bias", "decoder.decoder.8.conv2.weight", "decoder.decoder.8.conv2.bias", "decoder.decoder.8.conv_list.0.weight", "decoder.decoder.8.conv_list.0.bias", "decoder.decoder.8.conv_list.1.weight", "decoder.decoder.8.conv_list.1.bias", "decoder.decoder.8.last_conv.weight", "decoder.decoder.8.last_conv.bias", "decoder.decoder.8.bn1.weight", "decoder.decoder.8.bn1.bias", "decoder.decoder.8.bn1.running_mean", "decoder.decoder.8.bn1.running_var", "decoder.decoder.8.bn1.num_batches_tracked", "decoder.decoder.8.bn2.weight", "decoder.decoder.8.bn2.bias", "decoder.decoder.8.bn2.running_mean", "decoder.decoder.8.bn2.running_var", "decoder.decoder.8.bn2.num_batches_tracked", "decoder.decoder.8.bn_list.0.weight", "decoder.decoder.8.bn_list.0.bias", "decoder.decoder.8.bn_list.0.running_mean", "decoder.decoder.8.bn_list.0.running_var", "decoder.decoder.8.bn_list.0.num_batches_tracked", "decoder.decoder.8.bn_list.1.weight", "decoder.decoder.8.bn_list.1.bias", "decoder.decoder.8.bn_list.1.running_mean", "decoder.decoder.8.bn_list.1.running_var", "decoder.decoder.8.bn_list.1.num_batches_tracked", "decoder.decoder.9.conv1.weight", "decoder.decoder.9.conv1.bias", "decoder.decoder.9.conv2.weight", "decoder.decoder.9.conv2.bias", "decoder.decoder.9.conv_list.0.weight", "decoder.decoder.9.conv_list.0.bias", "decoder.decoder.9.conv_list.1.weight", "decoder.decoder.9.conv_list.1.bias", "decoder.decoder.9.last_conv.weight", "decoder.decoder.9.last_conv.bias", "decoder.decoder.9.bn1.weight", "decoder.decoder.9.bn1.bias", "decoder.decoder.9.bn1.running_mean", "decoder.decoder.9.bn1.running_var", "decoder.decoder.9.bn1.num_batches_tracked", "decoder.decoder.9.bn2.weight", "decoder.decoder.9.bn2.bias", "decoder.decoder.9.bn2.running_mean", "decoder.decoder.9.bn2.running_var", "decoder.decoder.9.bn2.num_batches_tracked", "decoder.decoder.9.bn_list.0.weight", "decoder.decoder.9.bn_list.0.bias", "decoder.decoder.9.bn_list.0.running_mean", "decoder.decoder.9.bn_list.0.running_var", "decoder.decoder.9.bn_list.0.num_batches_tracked", "decoder.decoder.9.bn_list.1.weight", "decoder.decoder.9.bn_list.1.bias", "decoder.decoder.9.bn_list.1.running_mean", "decoder.decoder.9.bn_list.1.running_var", "decoder.decoder.9.bn_list.1.num_batches_tracked", "decoder.decoder.10.conv1.weight", "decoder.decoder.10.conv1.bias", "decoder.decoder.10.conv2.weight", "decoder.decoder.10.conv2.bias", "decoder.decoder.10.conv_list.0.weight", "decoder.decoder.10.conv_list.0.bias", "decoder.decoder.10.conv_list.1.weight", "decoder.decoder.10.conv_list.1.bias", "decoder.decoder.10.last_conv.weight", "decoder.decoder.10.last_conv.bias", "decoder.decoder.10.bn1.weight", "decoder.decoder.10.bn1.bias", "decoder.decoder.10.bn1.running_mean", "decoder.decoder.10.bn1.running_var", "decoder.decoder.10.bn1.num_batches_tracked", "decoder.decoder.10.bn2.weight", "decoder.decoder.10.bn2.bias", "decoder.decoder.10.bn2.running_mean", "decoder.decoder.10.bn2.running_var", "decoder.decoder.10.bn2.num_batches_tracked", "decoder.decoder.10.bn_list.0.weight", "decoder.decoder.10.bn_list.0.bias", "decoder.decoder.10.bn_list.0.running_mean", "decoder.decoder.10.bn_list.0.running_var", "decoder.decoder.10.bn_list.0.num_batches_tracked", "decoder.decoder.10.bn_list.1.weight", "decoder.decoder.10.bn_list.1.bias", "decoder.decoder.10.bn_list.1.running_mean", "decoder.decoder.10.bn_list.1.running_var", "decoder.decoder.10.bn_list.1.num_batches_tracked", "decoder.decoder.11.conv1.weight", "decoder.decoder.11.conv1.bias", "decoder.decoder.11.conv2.weight", "decoder.decoder.11.conv2.bias", "decoder.decoder.11.conv_list.0.weight", "decoder.decoder.11.conv_list.0.bias", "decoder.decoder.11.conv_list.1.weight", "decoder.decoder.11.conv_list.1.bias", "decoder.decoder.11.last_conv.weight", "decoder.decoder.11.last_conv.bias", "decoder.decoder.11.bn1.weight", "decoder.decoder.11.bn1.bias", "decoder.decoder.11.bn1.running_mean", "decoder.decoder.11.bn1.running_var", "decoder.decoder.11.bn1.num_batches_tracked", "decoder.decoder.11.bn2.weight", "decoder.decoder.11.bn2.bias", "decoder.decoder.11.bn2.running_mean", "decoder.decoder.11.bn2.running_var", "decoder.decoder.11.bn2.num_batches_tracked", "decoder.decoder.11.bn_list.0.weight", "decoder.decoder.11.bn_list.0.bias", "decoder.decoder.11.bn_list.0.running_mean", "decoder.decoder.11.bn_list.0.running_var", "decoder.decoder.11.bn_list.0.num_batches_tracked", "decoder.decoder.11.bn_list.1.weight", "decoder.decoder.11.bn_list.1.bias", "decoder.decoder.11.bn_list.1.running_mean", "decoder.decoder.11.bn_list.1.running_var", "decoder.decoder.11.bn_list.1.num_batches_tracked", "decoder.decoder.12.conv1.weight", "decoder.decoder.12.conv1.bias", "decoder.decoder.12.conv2.weight", "decoder.decoder.12.conv2.bias", "decoder.decoder.12.conv_list.0.weight", "decoder.decoder.12.conv_list.0.bias", "decoder.decoder.12.conv_list.1.weight", "decoder.decoder.12.conv_list.1.bias", "decoder.decoder.12.last_conv.weight", "decoder.decoder.12.last_conv.bias", "decoder.decoder.12.bn1.weight", "decoder.decoder.12.bn1.bias", "decoder.decoder.12.bn1.running_mean", "decoder.decoder.12.bn1.running_var", "decoder.decoder.12.bn1.num_batches_tracked", "decoder.decoder.12.bn2.weight", "decoder.decoder.12.bn2.bias", "decoder.decoder.12.bn2.running_mean", "decoder.decoder.12.bn2.running_var", "decoder.decoder.12.bn2.num_batches_tracked", "decoder.decoder.12.bn_list.0.weight", "decoder.decoder.12.bn_list.0.bias", "decoder.decoder.12.bn_list.0.running_mean", "decoder.decoder.12.bn_list.0.running_var", "decoder.decoder.12.bn_list.0.num_batches_tracked", "decoder.decoder.12.bn_list.1.weight", "decoder.decoder.12.bn_list.1.bias", "decoder.decoder.12.bn_list.1.running_mean", "decoder.decoder.12.bn_list.1.running_var", "decoder.decoder.12.bn_list.1.num_batches_tracked", "decoder.decoder.13.conv1.weight", "decoder.decoder.13.conv1.bias", "decoder.decoder.13.conv2.weight", "decoder.decoder.13.conv2.bias", "decoder.decoder.13.conv_list.0.weight", "decoder.decoder.13.conv_list.0.bias", "decoder.decoder.13.conv_list.1.weight", "decoder.decoder.13.conv_list.1.bias", "decoder.decoder.13.last_conv.weight", "decoder.decoder.13.last_conv.bias", "decoder.decoder.13.bn1.weight", "decoder.decoder.13.bn1.bias", "decoder.decoder.13.bn1.running_mean", "decoder.decoder.13.bn1.running_var", "decoder.decoder.13.bn1.num_batches_tracked", "decoder.decoder.13.bn2.weight", "decoder.decoder.13.bn2.bias", "decoder.decoder.13.bn2.running_mean", "decoder.decoder.13.bn2.running_var", "decoder.decoder.13.bn2.num_batches_tracked", "decoder.decoder.13.bn_list.0.weight", "decoder.decoder.13.bn_list.0.bias", "decoder.decoder.13.bn_list.0.running_mean", "decoder.decoder.13.bn_list.0.running_var", "decoder.decoder.13.bn_list.0.num_batches_tracked", "decoder.decoder.13.bn_list.1.weight", "decoder.decoder.13.bn_list.1.bias", "decoder.decoder.13.bn_list.1.running_mean", "decoder.decoder.13.bn_list.1.running_var", "decoder.decoder.13.bn_list.1.num_batches_tracked", "decoder.decoder.14.conv1.weight", "decoder.decoder.14.conv1.bias", "decoder.decoder.14.conv2.weight", "decoder.decoder.14.conv2.bias", "decoder.decoder.14.conv_list.0.weight", "decoder.decoder.14.conv_list.0.bias", "decoder.decoder.14.conv_list.1.weight", "decoder.decoder.14.conv_list.1.bias", "decoder.decoder.14.last_conv.weight", "decoder.decoder.14.last_conv.bias", "decoder.decoder.14.bn1.weight", "decoder.decoder.14.bn1.bias", "decoder.decoder.14.bn1.running_mean", "decoder.decoder.14.bn1.running_var", "decoder.decoder.14.bn1.num_batches_tracked", "decoder.decoder.14.bn2.weight", "decoder.decoder.14.bn2.bias", "decoder.decoder.14.bn2.running_mean", "decoder.decoder.14.bn2.running_var", "decoder.decoder.14.bn2.num_batches_tracked", "decoder.decoder.14.bn_list.0.weight", "decoder.decoder.14.bn_list.0.bias", "decoder.decoder.14.bn_list.0.running_mean", "decoder.decoder.14.bn_list.0.running_var", "decoder.decoder.14.bn_list.0.num_batches_tracked", "decoder.decoder.14.bn_list.1.weight", "decoder.decoder.14.bn_list.1.bias", "decoder.decoder.14.bn_list.1.running_mean", "decoder.decoder.14.bn_list.1.running_var", "decoder.decoder.14.bn_list.1.num_batches_tracked", "decoder.decoder.15.conv1.weight", "decoder.decoder.15.conv1.bias", "decoder.decoder.15.conv2.weight", "decoder.decoder.15.conv2.bias", "decoder.decoder.15.conv_list.0.weight", "decoder.decoder.15.conv_list.0.bias", "decoder.decoder.15.conv_list.1.weight", "decoder.decoder.15.conv_list.1.bias", "decoder.decoder.15.last_conv.weight", "decoder.decoder.15.last_conv.bias", "decoder.decoder.15.bn1.weight", "decoder.decoder.15.bn1.bias", "decoder.decoder.15.bn1.running_mean", "decoder.decoder.15.bn1.running_var", "decoder.decoder.15.bn1.num_batches_tracked", "decoder.decoder.15.bn2.weight", "decoder.decoder.15.bn2.bias", "decoder.decoder.15.bn2.running_mean", "decoder.decoder.15.bn2.running_var", "decoder.decoder.15.bn2.num_batches_tracked", "decoder.decoder.15.bn_list.0.weight", "decoder.decoder.15.bn_list.0.bias", "decoder.decoder.15.bn_list.0.running_mean", "decoder.decoder.15.bn_list.0.running_var", "decoder.decoder.15.bn_list.0.num_batches_tracked", "decoder.decoder.15.bn_list.1.weight", "decoder.decoder.15.bn_list.1.bias", "decoder.decoder.15.bn_list.1.running_mean", "decoder.decoder.15.bn_list.1.running_var", "decoder.decoder.15.bn_list.1.num_batches_tracked", "decoder.decoder.16.conv1.weight", "decoder.decoder.16.conv1.bias", "decoder.decoder.16.conv2.weight", "decoder.decoder.16.conv2.bias", "decoder.decoder.16.conv_list.0.weight", "decoder.decoder.16.conv_list.0.bias", "decoder.decoder.16.conv_list.1.weight", "decoder.decoder.16.conv_list.1.bias", "decoder.decoder.16.last_conv.weight", "decoder.decoder.16.last_conv.bias", "decoder.decoder.16.bn1.weight", "decoder.decoder.16.bn1.bias", "decoder.decoder.16.bn1.running_mean", "decoder.decoder.16.bn1.running_var", "decoder.decoder.16.bn1.num_batches_tracked", "decoder.decoder.16.bn2.weight", "decoder.decoder.16.bn2.bias", "decoder.decoder.16.bn2.running_mean", "decoder.decoder.16.bn2.running_var", "decoder.decoder.16.bn2.num_batches_tracked", "decoder.decoder.16.bn_list.0.weight", "decoder.decoder.16.bn_list.0.bias", "decoder.decoder.16.bn_list.0.running_mean", "decoder.decoder.16.bn_list.0.running_var", "decoder.decoder.16.bn_list.0.num_batches_tracked", "decoder.decoder.16.bn_list.1.weight", "decoder.decoder.16.bn_list.1.bias", "decoder.decoder.16.bn_list.1.running_mean", "decoder.decoder.16.bn_list.1.running_var", "decoder.decoder.16.bn_list.1.num_batches_tracked", "decoder.decoder.17.conv1.weight", "decoder.decoder.17.conv1.bias", "decoder.decoder.17.conv2.weight", "decoder.decoder.17.conv2.bias", "decoder.decoder.17.conv_list.0.weight", "decoder.decoder.17.conv_list.0.bias", "decoder.decoder.17.conv_list.1.weight", "decoder.decoder.17.conv_list.1.bias", "decoder.decoder.17.last_conv.weight", "decoder.decoder.17.last_conv.bias", "decoder.decoder.17.bn1.weight", "decoder.decoder.17.bn1.bias", "decoder.decoder.17.bn1.running_mean", "decoder.decoder.17.bn1.running_var", "decoder.decoder.17.bn1.num_batches_tracked", "decoder.decoder.17.bn2.weight", "decoder.decoder.17.bn2.bias", "decoder.decoder.17.bn2.running_mean", "decoder.decoder.17.bn2.running_var", "decoder.decoder.17.bn2.num_batches_tracked", "decoder.decoder.17.bn_list.0.weight", "decoder.decoder.17.bn_list.0.bias", "decoder.decoder.17.bn_list.0.running_mean", "decoder.decoder.17.bn_list.0.running_var", "decoder.decoder.17.bn_list.0.num_batches_tracked", "decoder.decoder.17.bn_list.1.weight", "decoder.decoder.17.bn_list.1.bias", "decoder.decoder.17.bn_list.1.running_mean", "decoder.decoder.17.bn_list.1.running_var", "decoder.decoder.17.bn_list.1.num_batches_tracked", "decoder.decoder.18.conv1.weight", "decoder.decoder.18.conv1.bias", "decoder.decoder.18.conv2.weight", "decoder.decoder.18.conv2.bias", "decoder.decoder.18.conv_list.0.weight", "decoder.decoder.18.conv_list.0.bias", "decoder.decoder.18.conv_list.1.weight", "decoder.decoder.18.conv_list.1.bias", "decoder.decoder.18.last_conv.weight", "decoder.decoder.18.last_conv.bias", "decoder.decoder.18.bn1.weight", "decoder.decoder.18.bn1.bias", "decoder.decoder.18.bn1.running_mean", "decoder.decoder.18.bn1.running_var", "decoder.decoder.18.bn1.num_batches_tracked", "decoder.decoder.18.bn2.weight", "decoder.decoder.18.bn2.bias", "decoder.decoder.18.bn2.running_mean", "decoder.decoder.18.bn2.running_var", "decoder.decoder.18.bn2.num_batches_tracked", "decoder.decoder.18.bn_list.0.weight", "decoder.decoder.18.bn_list.0.bias", "decoder.decoder.18.bn_list.0.running_mean", "decoder.decoder.18.bn_list.0.running_var", "decoder.decoder.18.bn_list.0.num_batches_tracked", "decoder.decoder.18.bn_list.1.weight", "decoder.decoder.18.bn_list.1.bias", "decoder.decoder.18.bn_list.1.running_mean", "decoder.decoder.18.bn_list.1.running_var", "decoder.decoder.18.bn_list.1.num_batches_tracked", "decoder.decoder.19.conv1.weight", "decoder.decoder.19.conv1.bias", "decoder.decoder.19.conv2.weight", "decoder.decoder.19.conv2.bias", "decoder.decoder.19.conv_list.0.weight", "decoder.decoder.19.conv_list.0.bias", "decoder.decoder.19.conv_list.1.weight", "decoder.decoder.19.conv_list.1.bias", "decoder.decoder.19.last_conv.weight", "decoder.decoder.19.last_conv.bias", "decoder.decoder.19.bn1.weight", "decoder.decoder.19.bn1.bias", "decoder.decoder.19.bn1.running_mean", "decoder.decoder.19.bn1.running_var", "decoder.decoder.19.bn1.num_batches_tracked", "decoder.decoder.19.bn2.weight", "decoder.decoder.19.bn2.bias", "decoder.decoder.19.bn2.running_mean", "decoder.decoder.19.bn2.running_var", "decoder.decoder.19.bn2.num_batches_tracked", "decoder.decoder.19.bn_list.0.weight", "decoder.decoder.19.bn_list.0.bias", "decoder.decoder.19.bn_list.0.running_mean", "decoder.decoder.19.bn_list.0.running_var", "decoder.decoder.19.bn_list.0.num_batches_tracked", "decoder.decoder.19.bn_list.1.weight", "decoder.decoder.19.bn_list.1.bias", "decoder.decoder.19.bn_list.1.running_mean", "decoder.decoder.19.bn_list.1.running_var", "decoder.decoder.19.bn_list.1.num_batches_tracked", "decoder.decoder.20.conv1.weight", "decoder.decoder.20.conv1.bias", "decoder.decoder.20.conv2.weight", "decoder.decoder.20.conv2.bias", "decoder.decoder.20.conv_list.0.weight", "decoder.decoder.20.conv_list.0.bias", "decoder.decoder.20.conv_list.1.weight", "decoder.decoder.20.conv_list.1.bias", "decoder.decoder.20.last_conv.weight", "decoder.decoder.20.last_conv.bias", "decoder.decoder.20.bn1.weight", "decoder.decoder.20.bn1.bias", "decoder.decoder.20.bn1.running_mean", "decoder.decoder.20.bn1.running_var", "decoder.decoder.20.bn1.num_batches_tracked", "decoder.decoder.20.bn2.weight", "decoder.decoder.20.bn2.bias", "decoder.decoder.20.bn2.running_mean", "decoder.decoder.20.bn2.running_var", "decoder.decoder.20.bn2.num_batches_tracked", "decoder.decoder.20.bn_list.0.weight", "decoder.decoder.20.bn_list.0.bias", "decoder.decoder.20.bn_list.0.running_mean", "decoder.decoder.20.bn_list.0.running_var", "decoder.decoder.20.bn_list.0.num_batches_tracked", "decoder.decoder.20.bn_list.1.weight", "decoder.decoder.20.bn_list.1.bias", "decoder.decoder.20.bn_list.1.running_mean", "decoder.decoder.20.bn_list.1.running_var", "decoder.decoder.20.bn_list.1.num_batches_tracked", "decoder.decoder.21.conv1.weight", "decoder.decoder.21.conv1.bias", "decoder.decoder.21.conv2.weight", "decoder.decoder.21.conv2.bias", "decoder.decoder.21.conv_list.0.weight", "decoder.decoder.21.conv_list.0.bias", "decoder.decoder.21.conv_list.1.weight", "decoder.decoder.21.conv_list.1.bias", "decoder.decoder.21.last_conv.weight", "decoder.decoder.21.last_conv.bias", "decoder.decoder.21.bn1.weight", "decoder.decoder.21.bn1.bias", "decoder.decoder.21.bn1.running_mean", "decoder.decoder.21.bn1.running_var", "decoder.decoder.21.bn1.num_batches_tracked", "decoder.decoder.21.bn2.weight", "decoder.decoder.21.bn2.bias", "decoder.decoder.21.bn2.running_mean", "decoder.decoder.21.bn2.running_var", "decoder.decoder.21.bn2.num_batches_tracked", "decoder.decoder.21.bn_list.0.weight", "decoder.decoder.21.bn_list.0.bias", "decoder.decoder.21.bn_list.0.running_mean", "decoder.decoder.21.bn_list.0.running_var", "decoder.decoder.21.bn_list.0.num_batches_tracked", "decoder.decoder.21.bn_list.1.weight", "decoder.decoder.21.bn_list.1.bias", "decoder.decoder.21.bn_list.1.running_mean", "decoder.decoder.21.bn_list.1.running_var", "decoder.decoder.21.bn_list.1.num_batches_tracked", "decoder.decoder.22.conv1.weight", "decoder.decoder.22.conv1.bias", "decoder.decoder.22.conv2.weight", "decoder.decoder.22.conv2.bias", "decoder.decoder.22.conv_list.0.weight", "decoder.decoder.22.conv_list.0.bias", "decoder.decoder.22.conv_list.1.weight", "decoder.decoder.22.conv_list.1.bias", "decoder.decoder.22.last_conv.weight", "decoder.decoder.22.last_conv.bias", "decoder.decoder.22.bn1.weight", "decoder.decoder.22.bn1.bias", "decoder.decoder.22.bn1.running_mean", "decoder.decoder.22.bn1.running_var", "decoder.decoder.22.bn1.num_batches_tracked", "decoder.decoder.22.bn2.weight", "decoder.decoder.22.bn2.bias", "decoder.decoder.22.bn2.running_mean", "decoder.decoder.22.bn2.running_var", "decoder.decoder.22.bn2.num_batches_tracked", "decoder.decoder.22.bn_list.0.weight", "decoder.decoder.22.bn_list.0.bias", "decoder.decoder.22.bn_list.0.running_mean", "decoder.decoder.22.bn_list.0.running_var", "decoder.decoder.22.bn_list.0.num_batches_tracked", "decoder.decoder.22.bn_list.1.weight", "decoder.decoder.22.bn_list.1.bias", "decoder.decoder.22.bn_list.1.running_mean", "decoder.decoder.22.bn_list.1.running_var", "decoder.decoder.22.bn_list.1.num_batches_tracked", "decoder.decoder.23.conv1.weight", "decoder.decoder.23.conv1.bias", "decoder.decoder.23.conv2.weight", "decoder.decoder.23.conv2.bias", "decoder.decoder.23.conv_list.0.weight", "decoder.decoder.23.conv_list.0.bias", "decoder.decoder.23.conv_list.1.weight", "decoder.decoder.23.conv_list.1.bias", "decoder.decoder.23.last_conv.weight", "decoder.decoder.23.last_conv.bias", "decoder.decoder.23.bn1.weight", "decoder.decoder.23.bn1.bias", "decoder.decoder.23.bn1.running_mean", "decoder.decoder.23.bn1.running_var", "decoder.decoder.23.bn1.num_batches_tracked", "decoder.decoder.23.bn2.weight", "decoder.decoder.23.bn2.bias", "decoder.decoder.23.bn2.running_mean", "decoder.decoder.23.bn2.running_var", "decoder.decoder.23.bn2.num_batches_tracked", "decoder.decoder.23.bn_list.0.weight", "decoder.decoder.23.bn_list.0.bias", "decoder.decoder.23.bn_list.0.running_mean", "decoder.decoder.23.bn_list.0.running_var", "decoder.decoder.23.bn_list.0.num_batches_tracked", "decoder.decoder.23.bn_list.1.weight", "decoder.decoder.23.bn_list.1.bias", "decoder.decoder.23.bn_list.1.running_mean", "decoder.decoder.23.bn_list.1.running_var", "decoder.decoder.23.bn_list.1.num_batches_tracked", "decoder.decoder.24.conv1.weight", "decoder.decoder.24.conv1.bias", "decoder.decoder.24.conv2.weight", "decoder.decoder.24.conv2.bias", "decoder.decoder.24.conv_list.0.weight", "decoder.decoder.24.conv_list.0.bias", "decoder.decoder.24.conv_list.1.weight", "decoder.decoder.24.conv_list.1.bias", "decoder.decoder.24.last_conv.weight", "decoder.decoder.24.last_conv.bias", "decoder.decoder.24.bn1.weight", "decoder.decoder.24.bn1.bias", "decoder.decoder.24.bn1.running_mean", "decoder.decoder.24.bn1.running_var", "decoder.decoder.24.bn1.num_batches_tracked", "decoder.decoder.24.bn2.weight", "decoder.decoder.24.bn2.bias", "decoder.decoder.24.bn2.running_mean", "decoder.decoder.24.bn2.running_var", "decoder.decoder.24.bn2.num_batches_tracked", "decoder.decoder.24.bn_list.0.weight", "decoder.decoder.24.bn_list.0.bias", "decoder.decoder.24.bn_list.0.running_mean", "decoder.decoder.24.bn_list.0.running_var", "decoder.decoder.24.bn_list.0.num_batches_tracked", "decoder.decoder.24.bn_list.1.weight", "decoder.decoder.24.bn_list.1.bias", "decoder.decoder.24.bn_list.1.running_mean", "decoder.decoder.24.bn_list.1.running_var", "decoder.decoder.24.bn_list.1.num_batches_tracked". 
        size mismatch for decoder.decoder.0.conv1.weight: copying a param with shape torch.Size([1024, 2, 1]) from checkpoint, the shape in current model is torch.Size([1024, 3, 1]).

[kuwan2e]

Hello . I met the same problem. And I found how to solve it.
For autoencoder_25_squares, you need to use --dir_name {PATH_TO_autoencoder_25_squares}