The error points to different shapes of your parameters, which means that you’ve initialized the model in a different way.
Could you check, how you’ve initialized the ConvNet before saving the state_dict and use the same arguments?