Update train.py

chrischoy · Apr 23, 2020 · fc01d5b · fc01d5b
1 parent 416bd68
commit fc01d5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/train.py b/lib/train.py
@@ -80,7 +80,7 @@ def train(model, data_loader, val_data_loader, config, transform_data_fn=None):
         # Preprocess input
         color = input[:, :3].int()
         if config.normalize_color:
-          input[:, :3] = input[:, 1:] / 255. - 0.5
+          input[:, 1:] = input[:, 1:] / 255. - 0.5
         sinput = SparseTensor(input, coords).to(device)
 
         data_time += data_timer.toc(False)