Update train.py

This commit is contained in:
jinzr 2024-03-06 02:24:49 +08:00
parent ff430b465f
commit 6953980d4f