diff --git a/tests/utils/test_flops.py b/tests/utils/test_flops.py index f5533fa325..5b0272c9c6 100644 --- a/tests/utils/test_flops.py +++ b/tests/utils/test_flops.py @@ -136,7 +136,6 @@ def test_torch_linear_layer(self) -> None: def test_torch_pretrained_module(self) -> None: """Make sure FLOPs calculation works for a resnet18.""" - # pyre-fixme[16]: Module `models` has no attribute `resnet18`. mod = models.resnet18() inp = torch.randn(1, 3, 224, 224) with FlopTensorDispatchMode(mod) as ftdm: