Skip to content

Commit

Permalink
Fix for the error: RuntimeError: Expected object of scalar type Doubl…
Browse files Browse the repository at this point in the history
…e but got scalar type Float for argument fastai#3 'mat2' in call to _th_addmm_out. This error occurs in the notebook: lesson3-head-pose.ipynb. (fastai#2573)
  • Loading branch information
rraminen authored May 12, 2020
1 parent fa967e0 commit 8be0c6f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fastai/vision/image.py
Original file line number Diff line number Diff line change
Expand Up @@ -252,7 +252,7 @@ class ImagePoints(Image):
"Support applying transforms to a `flow` of points."
def __init__(self, flow:FlowField, scale:bool=True, y_first:bool=True):
if scale: flow = scale_flow(flow)
if y_first: flow.flow = flow.flow.flip(1)
if y_first: flow.flow = flow.flow.flip(1).float()
self._flow = flow
self._affine_mat = None
self.flow_func = []
Expand Down

0 comments on commit 8be0c6f

Please sign in to comment.