Skip to content

Commit d833d33

Browse files
NicolasHugfacebook-github-bot
authored andcommitted
[fbsync] Relative import for roi_align (#4533)
Reviewed By: kazhang Differential Revision: D31379420 fbshipit-source-id: 621ac9f856ca2619fa2d4b140ed1b95c699e722a
1 parent 4232ee4 commit d833d33

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/ops/poolers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from torch import nn, Tensor
33

44
import torchvision
5-
from torchvision.ops import roi_align
5+
from .roi_align import roi_align
66
from torchvision.ops.boxes import box_area
77

88
from typing import Optional, List, Dict, Tuple, Union

0 commit comments

Comments
 (0)