The spatial size of the input is too small for the model architecture, as one of the layers would create an empty tensor as given in the error message. Increase the spatial size of the input and it should work.