voxel: rebalance semantic gradnorm

d4l3k · Oct 6, 2023 · bf9415a · bf9415a
1 parent 1d4ecbd
commit bf9415a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchdrive/tasks/voxel.py b/torchdrive/tasks/voxel.py
@@ -1008,7 +1008,7 @@ def _semantic_loss(
             scales=3,
             mask=mask,
         )
-        sem_loss = sem_loss * per_pixel_weights * 100 * 1000 / 16
+        sem_loss = sem_loss * per_pixel_weights * 100 * 1000 / 32
 
         if ctx.log_text:
             pred_min, pred_max = semantic_classes.aminmax()