Closed ghost closed 4 years ago
Apparently you changed the batch and patch sizes to be 2 and 300x300 respectively, instead of 128 and 40x40 as the original implementation. Check that as you may have created some mismatches between shapes of y_true and y_pred when the loss is computed.
Thanks, I have found an alternative.
Hi @cszn -
I have been trying to use the DnCNN architecture - I am encountering the below error at Subtract layer. Can you suggest where I am heading wrong?
**InvalidArgumentError: Incompatible shapes: [2,300,300,1] vs. [2,2] [[node gradients/loss/subtract13_loss/sum_squared_error/sub_grad/BroadcastGradientArgs (defined at /opt/conda/lib/python3.6/site-packages/tensorflow_core/python/framework/ops.py:1751) ]] [Op:__inference_keras_scratch_graph_3338]
Function call stack: keras_scratch_graph**
Thanks