fix shape training

2025-06-26 16:03:44 +08:00
parent d48c432b58
commit 7c92655a0d
15 changed files with 199 additions and 657 deletions
--- a/hy3dshape/main.py
+++ b/hy3dshape/main.py
@@ -190,7 +190,7 @@ if __name__ == "__main__":
        precision=amp_type,
        callbacks=callbacks,
        accelerator="gpu",
-        devices=training_cfg.num_gpus,
+        devices=args.num_gpus,
        num_nodes=training_cfg.num_nodes,
        strategy=ddp_strategy,
        gradient_clip_val=training_cfg.get('gradient_clip_val'),