Merge pull request #52 from 0xsuu/master

Fixed a bug in look.py
daniilidis-group · May 22, 2019 · 5c7ff4f · 5c7ff4f
2 parents b4c60bc + 5de5bf1
commit 5c7ff4f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_renderer/look.py b/neural_renderer/look.py
@@ -17,7 +17,7 @@ def look(vertices, eye, direction=[0, 1, 0], up=None):
     elif isinstance(direction, np.ndarray):
         direction = torch.from_numpy(direction).to(device)
     elif torch.is_tensor(direction):
-        direction.to(device)
+        direction = direction.to(device)
 
     if isinstance(eye, list) or isinstance(eye, tuple):
         eye = torch.tensor(eye, dtype=torch.float32, device=device)