Skip to content

Commit

Permalink
suppress errors in torchtnt (#835)
Browse files Browse the repository at this point in the history
Summary: Pull Request resolved: #835

Differential Revision: D57559314

fbshipit-source-id: a8037e10b1a362eed267a2cd3b780745cd74e35a
  • Loading branch information
generatedunixname89002005307016 authored and facebook-github-bot committed May 20, 2024
1 parent 3e6fa3d commit 38cf06c
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion tests/utils/test_flops.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 38cf06c

Please sign in to comment.