I solved the size problem. but I have 6 classes in total. hence, he expects target efficiency to be at [10,6,256,256] dimensions. The output data is [10,2048,256,256] due to the resnet. How can I fix the output data?  I want it to be [10,6,256,256].