|
|
|
@ -20,7 +20,7 @@ def main():
|
|
|
|
|
|
|
|
|
|
adam_optimizer = paddle.optimizer.Adam(learning_rate=0.01)
|
|
|
|
|
|
|
|
|
|
trainer = paddle.trainer.SGD(topology=cost,
|
|
|
|
|
trainer = paddle.trainer.SGD(cost=cost,
|
|
|
|
|
parameters=parameters,
|
|
|
|
|
update_equation=adam_optimizer)
|
|
|
|
|
|
|
|
|
@ -28,7 +28,7 @@ def main():
|
|
|
|
|
if isinstance(event, paddle.event.EndIteration):
|
|
|
|
|
if event.batch_id % 1000 == 0:
|
|
|
|
|
result = trainer.test(reader=paddle.reader.batched(
|
|
|
|
|
paddle.dataset.mnist.test_creator(), batch_size=256))
|
|
|
|
|
paddle.dataset.mnist.test(), batch_size=256))
|
|
|
|
|
|
|
|
|
|
print "Pass %d, Batch %d, Cost %f, %s, Testing metrics %s" % (
|
|
|
|
|
event.pass_id, event.batch_id, event.cost, event.metrics,
|
|
|
|
|