From 38cf06cf56a7edc96158f4864b1c5a59405efdf4 Mon Sep 17 00:00:00 2001 From: generatedunixname89002005307016 Date: Mon, 20 May 2024 11:01:25 -0700 Subject: [PATCH] suppress errors in `torchtnt` (#835) Summary: Pull Request resolved: https://github.com/pytorch/tnt/pull/835 Differential Revision: D57559314 fbshipit-source-id: a8037e10b1a362eed267a2cd3b780745cd74e35a --- tests/utils/test_flops.py | 1 - 1 file changed, 1 deletion(-) 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: