It works with an input of e.g. [224,224,3], but it does not work with an input of e.g. [224,300,3].
As you told, if it works with an input of [224, 300, 3], why it doesn’t work in my pc?
I installed cuda 10.1