I add more convs but still want to use the pretrained model.
The error says:
Missing key(s) in state_dict:  "Missing key(s) in state_dict: "layer1.0.ae_module.fc.0.weight", "layer1.0.ae_module.fc.0.bias"