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)))