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

Fix SSGR

Browse Source
This commit is contained in:
James Betker 2020-09-20 19:07:15 -06:00
parent 17c569ea62
commit 53a5657850
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/StructuredSwitchedGenerator.py
View File

@ -159,7 +159,7 @@ class SSGr1(nn.Module):
self.init_temperature = init_temperature self.init_temperature = init_temperature
self.final_temperature_step = 10000 self.final_temperature_step = 10000
def forward(self, x, *args): def forward(self, x, embedding):
noise_stds = [] noise_stds = []
# The attention_maps debugger outputs <x>. Save that here. # The attention_maps debugger outputs <x>. Save that here.
self.lr = x.detach().cpu() self.lr = x.detach().cpu()