I have implemented the binned validation and it looks like the network struggles with predicting low values of k on the face of it.
The whole thing really stagnates before 1 epoch (that’s epoch’ 64 on the plot below) but strangely there’s some good fitting to say k=0.3 too so not entirely sure why it under-performs on some low k images but not others. To me this would suggest a problem with the dataset no? Or an under representation of the test set?

Validation_loss_017.png1642×836 144 KB