I've finally gotten my custom Tensorflow model compiled and running. The input size is 288x352x2, and the output size should be 72x88x2, but instead it is 456192 (which is 72x88x72??). Also, the first 72x88x2 values in the output are all NaN. I've tried all sorts of sizing/complexity changes like changing the input/output size to be smaller or a power of 2, changing the depth of convolutions to make the convolution ops smaller, but all I ever receive is NaN.
Any help would be greatly appreciated.
