Packages and Modules
- megatron.energon
AugmentTaskEncoder
BaseCoreDatasetFactory
BaseWebdatasetFactory
Batch
BatchDataset
BlendDataset
CaptioningSample
CaptioningWebdataset
ConcatDataset
Cooker
CrudeSample
CrudeWebdataset
DatasetLoader
DatasetLoaderInterface
DefaultDecoderWebdatasetFactory
DefaultGenericWebdatasetFactory
DefaultTaskEncoder
EpochizeDataset
FilterDataset
GcDataset
GroupBatchDataset
ImageClassificationSample
ImageClassificationWebdataset
ImageSample
ImageWebdataset
InterleavedSample
InterleavedWebdataset
IterMapDataset
JoinedWebdatasetFactory
LimitDataset
LogSampleDataset
MapDataset
Metadataset
MetadatasetV2
MixBatchDataset
MultiChoiceVQASample
MultiChoiceVQAWebdataset
OCRSample
OCRWebdataset
PackingDataset
RepeatDataset
Sample
SavableDataLoader
SavableDataset
ShuffleBufferDataset
SimilarityInterleavedSample
SimilarityInterleavedWebdataset
SkipSample
StandardWebdatasetFactory
TaskEncoder
TextSample
TextWebdataset
VQAOCRWebdataset
VQASample
VQAWebdataset
VidQASample
VidQAWebdataset
WorkerConfig
basic_sample_keys()
batch_list()
batch_pad_stack()
batch_stack()
concat_pad()
generic_batch()
generic_concat()
get_loader()
get_savable_loader()
get_train_dataset()
get_val_dataset()
get_val_datasets()
homogeneous_concat_mix()
load_dataset()
prepare_metadataset()
stateless()
Cooker
basic_sample_keys()