Update training_cli.py

[gessyoo]

Slight change to fix training won't start with 2 a6000s, even if only one enabled in cli. I've trained 3 models with the modded code, no issues.