Skip to content

Commit 8ecb64b

Browse files
authored
Update test_ops.py
1 parent 0b0edc6 commit 8ecb64b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/test_ops.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1504,9 +1504,9 @@ def test_lastlevel_maxpool2d(self):
15041504

15051505
# skip what FPN really dit, simply copy results from feature maps here.
15061506
results = feature_maps
1507-
1508-
results, names = extra_blocks(results, feature_maps, names)
1507+
15091508
expected_pooled_featuremap = F.max_pool2d(feature_maps[-1], 2, 2, 0)
1509+
results, names = extra_blocks(results, feature_maps, names)
15101510

15111511
tol = 1e-3 if feature_maps[-1].dtype is torch.half else 1e-5
15121512
torch.testing.assert_close(expected_pooled_featuremap, results[-1], atol=tol, rtol=tol)

0 commit comments

Comments
 (0)