asdf

train_hairstyles_parallel.py

@@ -8,8 +8,8 @@
   阶段C: 启动服务后串行 step3(回调生成ref材质) + step4(模板) + step5(预览)
 
 用法:
-  cd /home/xsl/change_hair/project/hair_service_sd
-  python /home/xsl/change_hair/train_hairstyles_parallel.py
+  cd /home/ubuntu/change_hair/project/hair_service_sd
+  python /home/ubuntu/change_hair/train_hairstyles_parallel.py
 """
 import os
 import sys
@@ -18,7 +18,7 @@ import json
 import subprocess
 from concurrent.futures import ThreadPoolExecutor, as_completed
 
-HAIR_SERVICE_DIR = "/home/xsl/change_hair/project/hair_service_sd"
+HAIR_SERVICE_DIR = "/home/ubuntu/change_hair/project/hair_service_sd"
 os.chdir(HAIR_SERVICE_DIR)
 sys.path.insert(0, HAIR_SERVICE_DIR)
 os.environ.setdefault("HF_HUB_OFFLINE", "1")
@@ -29,11 +29,11 @@ os.environ.setdefault("CUDA_VISIBLE_DEVICES", "0")
 # ====== 配置：本次要训练的发型 ======
 # (hair_id, input_dir, gender, template_img)
 HAIRSTYLES = [
-    ("chang_tuoyuan", "/home/xsl/change_hair/hair_type_images/chang_tuoyuan", "girl", "/home/xsl/change_hair/hair_type_images/chang_tuoyuan/chang_tuoyuan.jpg"),
-    ("chang_bolang",  "/home/xsl/change_hair/hair_type_images/chang_bolang",  "girl", "/home/xsl/change_hair/hair_type_images/chang_bolang/chang_bolang.jpg"),
-    ("chang_zhixian", "/home/xsl/change_hair/hair_type_images/chang_zhixian", "girl", "/home/xsl/change_hair/hair_type_images/chang_zhixian/chang_zhixian.jpg"),
-    ("chang_huaban",  "/home/xsl/change_hair/hair_type_images/chang_huaban",  "girl", "/home/xsl/change_hair/hair_type_images/chang_huaban/chang_huaban.jpg"),
-    ("chang_xinxing", "/home/xsl/change_hair/hair_type_images/chang_xinxing", "girl", "/home/xsl/change_hair/hair_type_images/chang_xinxing/chang_xinxing.jpg"),
+    ("chang_tuoyuan", "/home/ubuntu/change_hair/hair_type_images/chang_tuoyuan", "girl", "/home/ubuntu/change_hair/hair_type_images/chang_tuoyuan/chang_tuoyuan.jpg"),
+    ("chang_bolang",  "/home/ubuntu/change_hair/hair_type_images/chang_bolang",  "girl", "/home/ubuntu/change_hair/hair_type_images/chang_bolang/chang_bolang.jpg"),
+    ("chang_zhixian", "/home/ubuntu/change_hair/hair_type_images/chang_zhixian", "girl", "/home/ubuntu/change_hair/hair_type_images/chang_zhixian/chang_zhixian.jpg"),
+    ("chang_huaban",  "/home/ubuntu/change_hair/hair_type_images/chang_huaban",  "girl", "/home/ubuntu/change_hair/hair_type_images/chang_huaban/chang_huaban.jpg"),
+    ("chang_xinxing", "/home/ubuntu/change_hair/hair_type_images/chang_xinxing", "girl", "/home/ubuntu/change_hair/hair_type_images/chang_xinxing/chang_xinxing.jpg"),
 ]
 PARALLEL = 2  # LoRA 训练并发数
 
@@ -137,7 +137,7 @@ def main():
     log(f"阶段B完成: LoRA 训练完成 {len(completed)}/{len(step1_ok)}")
 
     # 写一个完成清单文件，供阶段C脚本读取
-    done_file = "/home/xsl/change_hair/project/logs/train_batch_done.json"
+    done_file = "/home/ubuntu/change_hair/project/logs/train_batch_done.json"
     with open(done_file, "w") as f:
         json.dump({"completed": sorted(completed), "all": ids}, f)
     log(f"已写入完成清单: {done_file}")