fix typo

microsoft · Lynazhang · Apr 8, 2022 · Mar 9, 2022 · Mar 9, 2022 · Mar 9, 2022
commit b5364eb32cea1eef318a7685b504ef6c7d1d3a13
diff --git a/nn_meter/predictor/prediction/extract_feature.py b/nn_meter/predictor/prediction/extract_feature.py
@@ -63,7 +63,7 @@ def get_predict_features(config):
         elif "pool" in op and "global" not in op:
             features = [inputh, cin, cout, ks, s]
         elif "global-pool" in op or "global-avgpool" in op or "gap" in op:
-            inputh = item["inputh"]
+            inputh = item["inputh"] if hasattr(item, "inputh") else 1
             cin = item["cin"]
             features = [inputh, cin]
         elif "channelshuffle" in op: