Remove temporary base_model change

2020-10-26 11:13:01 -06:00 · 2020-10-26 11:13:01 -06:00 · 1ce863849a
commit 1ce863849a
parent 54accfa693
1 changed files with 1 additions and 3 deletions
--- a/codes/models/base_model.py
+++ b/codes/models/base_model.py
@ -108,9 +108,7 @@ class BaseModel():

        load_net_clean = OrderedDict()  # remove unnecessary 'module.'
        for k, v in load_net.items():
-            if 'teco_recurrent_process' in k:
-                continue
-            elif k.startswith('module.'):
+            if k.startswith('module.'):
                load_net_clean[k[7:]] = v
            else:
                load_net_clean[k] = v