Latest model is now saved correctly

[sarthakpati]

Fixes #538

Proposed Changes

• ensuring latest model is written

Checklist

• I have read the CONTRIBUTING guide
• My PR is based from the current GaNDLF master
• Non-breaking change (does not break existing functionality): provide as many details as possible for any breaking change
• Function/class source code documentation added/updated
• Code has been blacked for style consistency
• If applicable, version information has been updated in GANDLF/version.py
• If adding a git submodule, add to list of exceptions for black styling in pyproject.toml file
• Usage documentation has been updated, if appropriate
• History has been updated, if appropriate
• Tests added or modified to cover the changes; if coverage is reduced, please give explanation
• If customized dependency installation is required (i.e., a separate pip install step is needed for PR to be functional), please ensure it is reflected in all the files that control the CI, namely: python-test.yml, and all docker files [1,2,3,4]

[github-actions]

MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅

[codecov]

Codecov Report

Merging #546 (c7df7b0) into master (2ae26e0) will increase coverage by 0.05%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##           master     #546      +/-   ##
==========================================
+ Coverage   93.83%   93.89%   +0.05%     
==========================================
  Files         109      109              
  Lines        6944     6944              
==========================================
+ Hits         6516     6520       +4     
+ Misses        428      424       -4     

Flag	Coverage Δ
unittests	93.89% <100.00%> (+0.05%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
GANDLF/compute/training_loop.py	83.47% <100.00%> (+1.27%)	⬆️
...NDLF/data/preprocessing/template_matching/utils.py	97.67% <0.00%> (+2.32%)	⬆️

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more