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

Fix loadSRG2 typo

Browse Source
This commit is contained in:
James Betker 2020-07-14 10:20:53 -06:00
parent 1e4083a35b
commit 4bcc409fc7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/archs/SwitchedResidualGenerator_arch.py
View File

@ -241,7 +241,7 @@ class ConfigurableSwitchedResidualGenerator2(nn.Module):
state_dict['switches.%i.switch.attention_norm.accumulator' % (i,)] = t_state['switches.%i.switch.attention_norm.accumulator' % (i,)]
state_dict['switches.%i.switch.attention_norm.accumulator_index' % (i,)] = t_state['switches.%i.switch.attention_norm.accumulator_index' % (i,)]
state_dict['switches.%i.switch.attention_norm.accumulator_filled' % (i,)] = t_state['switches.%i.switch.attention_norm.accumulator_filled' % (i,)]
super(DualOutputSRG, self).load_state_dict(state_dict, strict)
super(ConfigurableSwitchedResidualGenerator2, self).load_state_dict(state_dict, strict)
class Interpolate(nn.Module):