From 1d61132af7413155fedc197f72903d01c624bd01 Mon Sep 17 00:00:00 2001 From: generatedunixname89002005307016 Date: Fri, 10 May 2024 01:18:43 -0700 Subject: [PATCH] upgrade pyre version in `fbcode/vision` - batch 2 Differential Revision: D57183103 fbshipit-source-id: 7e2f42ddc6a1fa02abc27a451987d67a00264cbb --- fvcore/nn/print_model_statistics.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/fvcore/nn/print_model_statistics.py b/fvcore/nn/print_model_statistics.py index 15932e7..d361177 100644 --- a/fvcore/nn/print_model_statistics.py +++ b/fvcore/nn/print_model_statistics.py @@ -310,7 +310,11 @@ def _get_input_sizes(iterable: Iterable[Any]) -> List[Any]: # pyre-ignore[2,3] def flop_count_str( - flops: FlopCountAnalysis, activations: Optional[ActivationCountAnalysis] = None + # pyre-fixme[11]: Annotation `FlopCountAnalysis` is not defined as a type. + # pyre-fixme[11]: Annotation `ActivationCountAnalysis` is not defined as a type. + flops: FlopCountAnalysis, + # pyre-fixme[11]: Annotation `ActivationCountAnalysis` is not defined as a type. + activations: Optional[ActivationCountAnalysis] = None, ) -> str: """ Calculates the parameters and flops of the model with the given inputs