Commit d4c58688 authored by Zecheng He's avatar Zecheng He Committed by Facebook GitHub Bot
Browse files

Align_corners cannot be set when mode is nearest

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/211

Align_core cannot be set if the mode is nearest. Change to default None.

Reviewed By: wat3rBro

Differential Revision: D35681284

fbshipit-source-id: 23c57112e588c0b4ac5facfd61a7af0aa8a07ef0
parent 2235f180
...@@ -145,8 +145,6 @@ class KeypointRCNNConvUpsamplePredictorNoUpscale(nn.Module): ...@@ -145,8 +145,6 @@ class KeypointRCNNConvUpsamplePredictorNoUpscale(nn.Module):
self.out_channels = num_keypoints self.out_channels = num_keypoints
def forward(self, x): def forward(self, x):
x = layers.interpolate( x = layers.interpolate(x, scale_factor=(2, 2), mode="nearest")
x, scale_factor=(2, 2), mode="nearest", align_corners=False
)
x = self.kps_score_lowres(x) x = self.kps_score_lowres(x)
return x return x
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment