mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove temporary base_model change

Browse Source
This commit is contained in:
James Betker 2020-10-26 11:13:01 -06:00
parent 54accfa693
commit 1ce863849a
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/base_model.py
View File

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