upgrade pyre version in fbcode/vision - batch 2

Differential Revision: D55395614

fbshipit-source-id: 71677892b5d6f219f6df25b4efb51fb0f6b1441b

fvcore/nn/print_model_statistics.py

@@ -9,8 +9,13 @@
 import torch
 from torch import nn
 
+# pyre-fixme[21]: Could not find module `fvcore.nn.activation_count`.
 from .activation_count import ActivationCountAnalysis
+
+# pyre-fixme[21]: Could not find module `fvcore.nn.flop_count`.
 from .flop_count import FlopCountAnalysis
+
+# pyre-fixme[21]: Could not find module `fvcore.nn.parameter_count`.
 from .parameter_count import parameter_count
 
 

tests/test_activation_count.py

@@ -10,6 +10,8 @@
 
 import torch
 import torch.nn as nn
+
+# pyre-fixme[21]: Could not find module `fvcore.nn.activation_count`.
 from fvcore.nn.activation_count import activation_count, ActivationCountAnalysis
 from fvcore.nn.jit_handles import Handle
 from numpy import prod

tests/test_flop_count.py

@@ -9,6 +9,8 @@
 
 import torch
 import torch.nn as nn
+
+# pyre-fixme[21]: Could not find module `fvcore.nn.flop_count`.
 from fvcore.nn.flop_count import _DEFAULT_SUPPORTED_OPS, flop_count, FlopCountAnalysis
 from fvcore.nn.jit_handles import Handle
 from torch.autograd.function import Function

tests/test_jit_model_analysis.py

@@ -12,6 +12,8 @@
 
 import torch
 import torch.nn as nn
+
+# pyre-fixme[21]: Could not find module `fvcore.nn.flop_count`.
 from fvcore.nn.flop_count import FlopCountAnalysis
 from fvcore.nn.jit_analysis import JitModelAnalysis
 from fvcore.nn.jit_handles import addmm_flop_jit, conv_flop_jit, Handle, linear_flop_jit

tests/test_param_count.py

@@ -5,6 +5,7 @@
 
 import unittest
 
+# pyre-fixme[21]: Could not find module `fvcore.nn.parameter_count`.
 from fvcore.nn.parameter_count import parameter_count, parameter_count_table
 from torch import nn