Skip to content

Commit

Permalink
Migrate fbcode/vision/fair/detectron2/projects/DensePose to per-tar…
Browse files Browse the repository at this point in the history
…get checking

Reviewed By: rchen152

Differential Revision: D65570631

fbshipit-source-id: d41d1faa723e40c64d26f55dcb5d499940ce5eb0
  • Loading branch information
grievejia authored and facebook-github-bot committed Nov 7, 2024
1 parent 8d85329 commit 9131ce0
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion projects/DensePose/densepose/modeling/roi_heads/roi_head.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,10 @@ def __init__(self, cfg, input_shape: Dict[str, ShapeSpec], in_features):
for in_feature in self.in_features:
head_ops = []
head_length = max(
1, int(np.log2(feature_strides[in_feature]) - np.log2(self.common_stride))
1,
# pyre-fixme[6]: For 1st argument expected `Union[bytes, complex,
# float, int, generic, str]` but got `Optional[int]`.
int(np.log2(feature_strides[in_feature]) - np.log2(self.common_stride)),
)
for k in range(head_length):
conv = Conv2d(
Expand Down

0 comments on commit 9131ce0

Please sign in to comment.