[Unified Checkpoint] fix load missing keys

[DesmonDay]

PR types

Bug fixes

PR changes

Others

Description

Fix loading missing keys.

[paddle-bot]

Thanks for your contribution!

[codecov]

Codecov Report

Attention: Patch coverage is 0% with 2 lines in your changes missing coverage. Please review.

Project coverage is 53.01%. Comparing base (2b3d7bf) to head (9cea306).
Report is 220 commits behind head on develop.

Files with missing lines	Patch %	Lines
...nlp/trainer/unified_checkpoint/check_completion.py	0.00%	2 Missing ⚠️

Additional details and impacted files

@@             Coverage Diff             @@
##           develop    #9523      +/-   ##
===========================================
+ Coverage    52.82%   53.01%   +0.18%     
===========================================
  Files          701      701              
  Lines       110594   110709     +115     
===========================================
+ Hits         58422    58691     +269     
+ Misses       52172    52018     -154     

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

[wawltor]

LGTM