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

Fix loss indexing

Browse Source
This commit is contained in:
James Betker 2020-10-09 20:20:51 -06:00
parent 202eb11fdc
commit 0011d445c8
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/steps/losses.py
View File

@ -413,5 +413,5 @@ class ForElementLoss(ConfigurableLoss):
def forward(self, net, state): def forward(self, net, state):
st = state.copy() st = state.copy()
st['_real'] = state[self.opt['real']][:, self.index] st['_real'] = state[self.opt['real']][:, self.index]
st['_fake'] = state[self.opt['fake']][:, self.index] st['_fake'] = state[self.opt['fake']][self.index]
return self.loss(net, st) return self.loss(net, st)