test: use batch loss and not average loss

samedii · samedii · commit 79979c20a5c7 · 2020-12-09T19:18:06.000+01:00
diff --git a/test/test_mnist.py b/test/test_mnist.py
@@ -57,7 +57,7 @@ def test_mnist():
         name="gradient",
         tensorboard_logger=tensorboard_logger,
         metrics=dict(
-            loss=lantern.MapMetric(lambda examples, predictions, loss: loss),
+            loss=lantern.ReduceMetric(lambda state, examples, predictions, loss: loss),
         ),
     )
 

Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,7 @@ def test_mnist():`
`57`	`57`	`name="gradient",`
`58`	`58`	`tensorboard_logger=tensorboard_logger,`
`59`	`59`	`metrics=dict(`
`60`		`- loss=lantern.MapMetric(lambda examples, predictions, loss: loss),`
	`60`	`+ loss=lantern.ReduceMetric(lambda state, examples, predictions, loss: loss),`
`61`	`61`	`),`
`62`	`62`	`)`
`63`	`63`