Channel should be 2048 if directly using the last stage in resnet-50, but height and width should be correlated with your input image’s resolution. The stride between input and the last feature map is 32, which means if your input size is 224 * 224 * 3, then the output (head1) should be 7 * 7 * 2048