From e60fe1df22fa9cd37003c258474b2aa1b8cde500 Mon Sep 17 00:00:00 2001 From: Yi Wang Date: Thu, 9 Feb 2017 18:56:51 -0800 Subject: [PATCH] Cost as a parameter to train and test --- doc/design/api.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/doc/design/api.md b/doc/design/api.md index 17ab50fdb2..dcf19451b1 100644 --- a/doc/design/api.md +++ b/doc/design/api.md @@ -10,13 +10,13 @@ Model = {topology, parameters} Evaluator = {Model*, activations} - forward -- test +- test(cost, ...) GradientMachine = {Evaluator*, gradients} - backward Optimizer = {Model*, GradientMachine*} -- train +- train(cost, ...) - update - checkpoint ``` -- GitLab