Uses of Interface
io.github.kirstenali.deepj.data.BatchSource

Packages that use BatchSource

Package	Description
io.github.kirstenali.deepj.data
io.github.kirstenali.deepj.training

Uses of BatchSource in io.github.kirstenali.deepj.data

Classes in io.github.kirstenali.deepj.data that implement BatchSource

Modifier and Type	Class	Description
final class	RandomAccessTextDataset	Random, bounded-memory batches sampled directly from a UTF-8 text file.
final class	TextDataset	Dataset that samples random contiguous chunks from token ids.

Uses of BatchSource in io.github.kirstenali.deepj.training

Methods in io.github.kirstenali.deepj.training with parameters of type BatchSource

Modifier and Type	Method	Description
static EvaluationResult	CausalLMEvaluation.evaluate(CausalLM model, BatchSource source, int batches, int batchSize)
static Trainer	CausalLMTraining.trainer(CausalLM model, BatchSource dataset, float lr)
static Trainer	CausalLMTraining.trainer(CausalLM model, BatchSource dataset, ParameterOptimizer optimizer)