提交 2d07a0bc 编写于 作者: Y yukavio

fix slim prune

上级 c58fb6e8
...@@ -92,7 +92,8 @@ def main(): ...@@ -92,7 +92,8 @@ def main():
sen = load_sensitivities("sensitivities_0.data") sen = load_sensitivities("sensitivities_0.data")
for i in skip_list: for i in skip_list:
sen.pop(i) if i in sen.keys():
sen.pop(i)
back_bone_list = ['conv' + str(x) for x in range(1, 5)] back_bone_list = ['conv' + str(x) for x in range(1, 5)]
for i in back_bone_list: for i in back_bone_list:
for key in list(sen.keys()): for key in list(sen.keys()):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册