fix #33

chrischoy · Apr 24, 2020 · 0b3ad50 · 0b3ad50
1 parent fc01d5b
commit 0b3ad50
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/train.py b/lib/train.py
@@ -74,8 +74,8 @@ def train(model, data_loader, val_data_loader, config, transform_data_fn=None):
         data_timer.tic()
         coords, input, target = data_iter.next()
 
-        # For some networks, making the network invariant to even, odd coords is important
-        coords[:, :3] += (torch.rand(3) * 100).type_as(coords)
+        # For some networks, making the network invariant to even, odd coords is important. Random translation
+        coords[:, 1:] += (torch.rand(3) * 100).type_as(coords)
 
         # Preprocess input
         color = input[:, :3].int()