OptimizationStrategy is the interface implemented by AdaGrad, Adam, etc.
| 14 | |
| 15 | // OptimizationStrategy is the interface implemented by AdaGrad, Adam, etc. |
| 16 | type OptimizationStrategy interface { |
| 17 | OptimizeParams(*nn.Param) error |
| 18 | } |
| 19 | |
| 20 | // Optimizer is an optimizer that can optimize a set of parameters. |
| 21 | type Optimizer struct { |