优化训练过程

2025-03-07 08:34:50 +08:00
parent 3d39e89e26
commit 865b077a91
3 changed files with 13 additions and 3 deletions
--- a/app/service/project_train_service.py
+++ b/app/service/project_train_service.py
@ -188,7 +188,9 @@ def run_commend(data: str, project: str,
             "--data=" + data,
             "--project=" + project,
             "--name=" + name,
-             "--epochs=" + str(epochs)],
+             "--epochs=" + str(epochs),
+             "--batch-size=4",
+             "--exist_ok"],
            bufsize=1,  # bufsize=0时，为不缓存；bufsize=1时，按行缓存；bufsize为其他正整数时，为按照近似该正整数的字节数缓存
            shell=False,
            stdout=subprocess.PIPE,