Skip to content

Commit 52d938f

Browse files
varunagrawalfmassa
authored andcommitted
Consistent naming for rois variable (#627)
* remove .type().tensor() calls in favor of the new approach to tensor initialization * Consistent naming for rois variable in ROIPool
1 parent 5092606 commit 52d938f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

torchvision/layers/roi_pool.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,14 +11,14 @@
1111

1212
class _ROIPool(Function):
1313
@staticmethod
14-
def forward(ctx, input, roi, output_size, spatial_scale):
14+
def forward(ctx, input, rois, output_size, spatial_scale):
1515
ctx.output_size = _pair(output_size)
1616
ctx.spatial_scale = spatial_scale
1717
ctx.input_shape = input.size()
1818
output, argmax = _C.roi_pool_forward(
19-
input, roi, spatial_scale,
19+
input, rois, spatial_scale,
2020
output_size[0], output_size[1])
21-
ctx.save_for_backward(roi, argmax)
21+
ctx.save_for_backward(rois, argmax)
2222
return output
2323

2424
@staticmethod

0 commit comments

Comments
 (0)