x = PReLU()(x)
This type of description does not work for variable input image size. Please use the below code instead.
x = PReLU(shared_axes=[1,2])(x)
x = PReLU()(x)
This type of description does not work for variable input image size. Please use the below code instead.
x = PReLU(shared_axes=[1,2])(x)