From 7d09bd849e8b12f29c3710c7ee150b1a22a747d2 Mon Sep 17 00:00:00 2001
From: SouthRiver <52597530+South-River@users.noreply.github.com>
Date: Wed, 28 May 2025 13:10:16 +0800
Subject: [PATCH] Update main_cuda.py

---
 main_cuda.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main_cuda.py b/main_cuda.py
index 2911b85..4928763 100644
--- a/main_cuda.py
+++ b/main_cuda.py
@@ -210,7 +210,7 @@ def is_save_iter(i):
         args.coef_speed * loss_speed + \
         args.coef_v_pred * loss_v_pred + \
         args.coef_collide * loss_collide + \
-        args.coef_ground_affinity + loss_ground_affinity
+        args.coef_ground_affinity * loss_ground_affinity
 
     if torch.isnan(loss):
         print("loss is nan, exiting...")