Skip to content

Commit

Permalink
typo fixed
Browse files Browse the repository at this point in the history
  • Loading branch information
zhen-he committed Sep 23, 2019
1 parent c242963 commit 92cd55d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion modules/loss_calculator.py
Expand Up @@ -25,7 +25,7 @@ def forward(self, output, target, area, **kwargs):
if 'Y_b' in kwargs.keys():
Y_b = kwargs['Y_b']
loss_recon = self.mse(output, target) + self.mse((output - Y_b).abs(), (target - Y_b).abs())
else
else:
loss_recon = self.mse(output, target)
loss = loss_recon
losses['recon'] = loss_recon.data[0]
Expand Down

0 comments on commit 92cd55d

Please sign in to comment.