Fix label datatype in TF Trainer
# What does this PR do?

This PR fixes the case where `labels` can be either a `dict` or a `tf.Tensor` when doing gradient accumulation.

I agree with Sylvain that while this is not tested, it's hard to recommend using it.