Skip to content

Commit d086e02

Browse files
committed
Fix docstyle
1 parent 17d885b commit d086e02

File tree

1 file changed

+2
-2
lines changed
  • pytorch_toolbelt/inference

1 file changed

+2
-2
lines changed

pytorch_toolbelt/inference/tta.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def fivecrop_image2label(model: nn.Module, image: Tensor,
7171
center_crop_x = (image_width - crop_width) // 2
7272

7373
crop_cc = image[..., center_crop_y:center_crop_y + crop_height,
74-
center_crop_x:center_crop_x + crop_width]
74+
center_crop_x:center_crop_x + crop_width]
7575
assert crop_cc.size(2) == crop_height
7676
assert crop_cc.size(3) == crop_width
7777

@@ -118,7 +118,7 @@ def tencrop_image2label(model: nn.Module, image: Tensor,
118118
center_crop_x = (image_width - crop_width) // 2
119119

120120
crop_cc = image[..., center_crop_y:center_crop_y + crop_height,
121-
center_crop_x:center_crop_x + crop_width]
121+
center_crop_x:center_crop_x + crop_width]
122122
assert crop_cc.size(2) == crop_height
123123
assert crop_cc.size(3) == crop_width
124124

0 commit comments

Comments
 (0)