How to change the input image size

[cuongnc220592]

I have a question for changing the size of input image. As the configuration, the input size is 511x511. However, I want to resize the input image to 1024x768 or 1024x1024. I have tried to change the config file "input_size": [1023, 1023].

But I have a error IndexError: The shape of the mask [1, 80, 128, 128] at index 2 does not match the shape of the indexed tensor [1, 80, 256, 256] at index 2

How can I solve the problem Thank you very much.