Testing the model

[Awenjie10]

Thank you for your code. When I test the pre-trained model with my own images, it appeared a problem, 'Missing key(s) in state_dict: "prior_conv1.weight", "prior_conv1.bias". How can I fix the problem? Thank you .