iicd/MeCo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
HamsterMimi
2023-05-04 13:23:56 +08:00
parent 189df25fd3
commit fd43e67da1
8 changed files with 1 additions and 211 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
zerocostnas/post_validate.py
View File

@@ -223,20 +223,6 @@ def main():
else:
#score = score_loop(network, None, train_queue, args.gpu, None, args.proj_crit)
network.requires_feature = False
if args.proj_crit == 'comb':
synflow = predictive.find_measures(network,
train_queue,
('random', 1, n_classes),
torch.device("cuda"),
measure_names=['synflow'])
var = predictive.find_measures(network,
train_queue,
('random', 1, n_classes),
torch.device("cuda"),
measure_names=['var'])
comb = np.log(synflow['synflow'] + 1) / (var['var'] + 0.1)
measures = {'comb': comb}
else:
measures = predictive.find_measures(network,
train_queue,