fix(train): fix checkpoint not properly loaded

[34j]

Closes #270

[codecov-commenter]

Codecov Report

Merging #271 (7231fc0) into main (13d6346) will decrease coverage by 0.03%.
The diff coverage is 12.82%.

❗ Current head 7231fc0 differs from pull request most recent head b83f0df. Consider uploading reports for the commit b83f0df to get more accurate results

📣 This organization is not using Codecov’s GitHub App Integration. We recommend you install it so Codecov can continue to function properly for your repositories. Learn more

@@            Coverage Diff             @@
##             main     #271      +/-   ##
==========================================
- Coverage   20.53%   20.51%   -0.03%     
==========================================
  Files          38       38              
  Lines        3224     3247      +23     
  Branches      415      418       +3     
==========================================
+ Hits          662      666       +4     
- Misses       2545     2564      +19     
  Partials       17       17              

Impacted Files	Coverage Δ
src/so_vits_svc_fork/train.py	22.84% <12.82%> (-0.73%)	⬇️

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