Commit 874481f2 authored by Michael Galkov's avatar Michael Galkov Committed by Soumith Chintala
Browse files

Fix typo (#158)

parent 1f2c15f2
...@@ -44,7 +44,7 @@ class ToTensor(object): ...@@ -44,7 +44,7 @@ class ToTensor(object):
if isinstance(pic, np.ndarray): if isinstance(pic, np.ndarray):
# handle numpy array # handle numpy array
img = torch.from_numpy(pic.transpose((2, 0, 1))) img = torch.from_numpy(pic.transpose((2, 0, 1)))
# backard compability # backward compatibility
return img.float().div(255) return img.float().div(255)
if accimage is not None and isinstance(pic, accimage.Image): if accimage is not None and isinstance(pic, accimage.Image):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment