import { default as Model, ModelForwardAttributes } from '../models/model';
import { default as BasicTrainer } from './BasicTrainer';
import { ITokeniser } from '../tokeniser/type';
import { DatasetBuilder } from './DatasetBuilder';
import { AdamWOptimizer } from './AdamW';
import { AdamWOptimizerConfig } from './types';
export default class PreTrainer extends BasicTrainer {
    tokenizer: ITokeniser;
    datasetBuilder: DatasetBuilder;
    constructor(model: Model<ModelForwardAttributes>, tokenizer: ITokeniser, optConfig?: Partial<AdamWOptimizerConfig>, optimizer?: AdamWOptimizer);
}
