[docs] Update the training snippets for some losses that should use the v3 Trainer

[docs] Update the training snippets for some losses that should use the v3 Trainer #1021