Ok - so thanks for writing that out, it helps.
Basically my model is specified output as [y,1] (pred here) so I have to specify the other input to the loss function also as a [y,1] instead of just [y].
Is that a fair explanation?