Update train_util.py

fix mistake
This commit is contained in:
TingTingin
2023-05-23 02:00:33 -04:00
committed by GitHub
parent 5a1a14f9fc
commit d859a3a925