【AutoParallel】Support bf16 loss in static

[heavyrain-lzy]

PR types

Others

PR changes

Others

Description

Support to log loss with dtype bfloat16 aligning to dygraph

PaddleNLP/paddlenlp/trainer/trainer.py

Line 2649 in 2a8d138

metrics[f"{metric_key_prefix}_loss"] = all_losses.mean().item()

[paddle-bot]

Thanks for your contribution!

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (4a9c766) 56.69% compared to head (be64171) 56.69%.
Report is 1 commits behind head on develop.

Additional details and impacted files

@@           Coverage Diff            @@
##           develop    #7874   +/-   ##
========================================
  Coverage    56.69%   56.69%           
========================================
  Files          588      588           
  Lines        89315    89315           
========================================
  Hits         50641    50641           
  Misses       38674    38674           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[zhiqiu]

LGTM