update

2023-05-14 10:57:08 +08:00
parent a37e99a057
commit 993d55076e
4 changed files with 281 additions and 1 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', 'meco', 'zico']:
+        if k in ['jacob_cov', 'var', 'cor', 'norm', 'meco', 'zico', 'ntk', 'gradsign', 'zen']:
            measures[k] = v
        else:
            measures[k] = sum_arr(v)