Hi,
The problem occurs due to the input format that you are using. You have to feed input in the following format - BxCxHxW. You have used the channel as the last dimension(BxHxWxC - this is wrong).
Thanks