I have this model in pytorch, which basically is a Multi-scale CNN model which returns the feature maps concatenated in the end for that image. I\'m not really sure how to write