Skip to content

Commit

Permalink
upgrade pyre version in fbcode/vision - batch 2
Browse files Browse the repository at this point in the history
Reviewed By: bottler

Differential Revision: D60992234

fbshipit-source-id: 899db6ed590ef966ff651c11027819e59b8401a3
  • Loading branch information
generatedunixname89002005307016 authored and facebook-github-bot committed Aug 9, 2024
1 parent 1d61132 commit f3b07ea
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/test_jit_model_analysis.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
x = x.reshape(-1, 2, 5)
x = self.conv(x)
x = torch.flatten(x, 1)
# pyre-fixme[9]: x has type `Tensor`; used as `int`.
x = 3 * self.fc(x) + 1
return x

Expand Down

0 comments on commit f3b07ea

Please sign in to comment.