import { Module } from '../module'; import { Optimizer } from './optimizer'; /** * Stochastic gradient descent optimizer */ export declare class SGD extends Optimizer { lr: number; /** * New SGD optimizer for a particular model. * * @param lr Learning rate, the step size for each update step */ constructor(model: Module, lr?: number); step(): void; }