Fix load of model checkpoints in the Trainer
# What does this PR do?

#18221 broke the model reload when the contributor removed the `strict_load` variable (as requested in the review) without setting it to its proper value in the calls to `load_state_dict` after. This PR addresses that.

Fixes #18373
_The documentation is not available anymore as the PR was closed or merged._