fix slim prune

2d07a0bc · yukavio · c58fb6e8 · 2d07a0bc
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

deploy/slim/prune/pruning_and_finetune.py deploy/slim/prune/pruning_and_finetune.py +2 -1

未找到文件。
--- a/deploy/slim/prune/pruning_and_finetune.py
+++ b/deploy/slim/prune/pruning_and_finetune.py
@@ -92,6 +92,7 @@ def main():

    sen = load_sensitivities("sensitivities_0.data")
    for i in skip_list:
+        if i in sen.keys():
            sen.pop(i)
    back_bone_list = ['conv' + str(x) for x in range(1, 5)]
    for i in back_bone_list: