- sign missing from loss function definition

2025-04-19 05:16:06 +00:00 · 2019-10-17 16:00:35 -07:00 · 2019-10-17 16:00:35 -07:00 · 1090e89a86
commit 1090e89a86
parent 1603a7393a
1 changed files with 1 additions and 1 deletions
--- a/examples/resnet50.py
+++ b/examples/resnet50.py
@ -101,7 +101,7 @@ if __name__ == "__main__":
  def loss(params, batch):
    inputs, targets = batch
    logits = predict_fun(params, inputs)
-    return np.sum(logits * targets)
+    return -np.sum(logits * targets)

  def accuracy(params, batch):
    inputs, targets = batch