From 1ce863849ad705623315330ea208e8b1e759a592 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 26 Oct 2020 11:13:01 -0600 Subject: [PATCH] Remove temporary base_model change --- codes/models/base_model.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/codes/models/base_model.py b/codes/models/base_model.py index 47664e8a..ea08aecc 100644 --- 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