Content-Length: 519 | pFad | http://github.com/speechbrain/speechbrain/pull/3033.diff
thub.com diff --git a/speechbrain/nnet/losses.py b/speechbrain/nnet/losses.py index fcf160edc9..c198286355 100644 --- a/speechbrain/nnet/losses.py +++ b/speechbrain/nnet/losses.py @@ -419,7 +419,7 @@ def classification_error( def error(predictions, targets): """Computes the classification error.""" - predictions = torch.argmax(probabilities, dim=-1) + predictions = torch.argmax(predictions, dim=-1) return (predictions != targets).float() return compute_masked_loss(Fetched URL: http://github.com/speechbrain/speechbrain/pull/3033.diff
Alternative Proxies: