I don’t know what kind of outputs your model provides, i.e. if they are logits, probabilities, class predictions etc., but outputs_ext_np.round() looks a bit unusual, so I would recommend to double check that it’s indeed working as expected.