update

2023-05-04 13:23:56 +08:00
parent 189df25fd3
commit fd43e67da1
8 changed files with 1 additions and 211 deletions
--- a/zero-cost-nas/foresight/pruners/predictive.py
+++ b/zero-cost-nas/foresight/pruners/predictive.py
@@ -108,7 +108,7 @@ def find_measures(net_orig,                  # neural network

    measures = {}
    for k,v in measures_arr.items():
-        if k in ['jacob_cov', 'var', 'cor', 'norm', 'meco', 'zico']:
+        if k in ['jacob_cov', 'meco', 'zico']:
            measures[k] = v
        else:
            measures[k] = sum_arr(v)