Interface BatchSource

All Known Implementing Classes:
RandomAccessTextDataset, TextDataset
Functional Interface:
This is a functional interface and can therefore be used as the assignment target for a lambda expression or method reference.

@FunctionalInterface public interface BatchSource
Supplies token batches to a causal language-model trainer.
  • Method Summary

    Modifier and Type
    Method
    Description
    nextBatch(int batchSize)
     
  • Method Details

    • nextBatch

      Batch nextBatch(int batchSize)