@hoff97/tensor-js/dist/lib/model/optimizer/SGD.d.ts

PyTorch like deep learning inferrence library

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;
}