From 4e3ef3e74d643de973e760156f144c3179edfa76 Mon Sep 17 00:00:00 2001
From: kinghuin <kinghuin_chull@163.com>
Date: Thu, 15 Oct 2020 10:20:36 +0800
Subject: [PATCH] Update strategy.py

fix https://github.com/PaddlePaddle/PaddleHub/issues/527
---
 paddlehub/finetune/strategy.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/paddlehub/finetune/strategy.py b/paddlehub/finetune/strategy.py
index 7f522c22..48f59d9b 100644
--- a/paddlehub/finetune/strategy.py
+++ b/paddlehub/finetune/strategy.py
@@ -102,6 +102,8 @@ def get_depth_parameter(main_program):
     depth_list = sorted(depth_params_dict.keys())
     len_depth_list = len(depth_list)
     for index, depth in enumerate(depth_list):
+        if depth not in updated_depth_params_dict:
+            continue
         for param in depth_params_dict[depth]:
             prefix = param.name.split(".")[0]
             if index < len_depth_list - 1:
-- 
GitLab