From 77286f6ce6a0e645814b5f9beb06476fffb11977 Mon Sep 17 00:00:00 2001
From: Zhipeng Xue <543984341@qq.com>
Date: Thu, 6 Jul 2023 17:08:19 +0800
Subject: [PATCH] Fix memory issue

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

diff --git a/train_semseg.py b/train_semseg.py
index fd1df5f88..305133f45 100644
--- a/train_semseg.py
+++ b/train_semseg.py
@@ -199,7 +199,7 @@ def bn_momentum_adjust(m, momentum):
             correct = np.sum(pred_choice == batch_label)
             total_correct += correct
             total_seen += (BATCH_SIZE * NUM_POINT)
-            loss_sum += loss
+            loss_sum += loss.item()
         log_string('Training mean loss: %f' % (loss_sum / num_batches))
         log_string('Training accuracy: %f' % (total_correct / float(total_seen)))