Type Alias TrainType

type TrainType = {
    beta1: number;
    beta2: number;
    blockSize: number;
    commentNthStep: number;
    docs: string[];
    eps: number;
    learningRate: number;
    nEmbd: number;
    nHead: number;
    nLayer: number;
    numSteps: number;
    onStep?: (
        update: {
            loss: number;
            numSteps: number;
            samples: string[];
            step: number;
        },
    ) => void;
    sampleCount: number;
    seed: number;
    signal?: { aborted: boolean };
    temperature: number;
}