fix(train): fix ddp not working

[34j]

Closes #287

[codecov-commenter]

Codecov Report

Merging #297 (31468eb) into main (d3ab7d6) will not change coverage.
The diff coverage is n/a.

📣 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     #297   +/-   ##
=======================================
  Coverage   20.46%   20.46%           
=======================================
  Files          38       38           
  Lines        3260     3260           
  Branches      420      420           
=======================================
  Hits          667      667           
  Misses       2576     2576           
  Partials       17       17           

Impacted Files	Coverage Δ
src/so_vits_svc_fork/train.py	21.90% <ø> (ø)

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