Skip to content
Snippets Groups Projects

Fixed criterion call

Closed Lupo Pasini, Massimiliano requested to merge fixed_criterion_call into master
2 files
+ 2
2
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -104,7 +104,7 @@ class Optimization:
if torch.is_grad_enabled():
self.optimizer.zero_grad()
output = self.network.forward(inputs)
loss = self.criterion(output, targets)
loss = criterion(output, targets)
if loss.requires_grad:
loss.backward()
return loss
Loading