Skip to content

Commit e60fe1d

Browse files
author
Yi Wang
committed
Cost as a parameter to train and test
1 parent 08c0051 commit e60fe1d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

doc/design/api.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,13 +10,13 @@ Model = {topology, parameters}
1010
1111
Evaluator = {Model*, activations}
1212
- forward
13-
- test
13+
- test(cost, ...)
1414
1515
GradientMachine = {Evaluator*, gradients}
1616
- backward
1717
1818
Optimizer = {Model*, GradientMachine*}
19-
- train
19+
- train(cost, ...)
2020
- update
2121
- checkpoint
2222
```

0 commit comments

Comments
 (0)