Index

optimus_dl.modules.data

DataPipeline

Bases: NamedTuple

DataPipeline(datasets, dataloader)

Parameters:

Name	Type	Description	Default
datasets	BaseNode		None
dataloader	BaseNode | Loader		None

Source code in optimus_dl/modules/data/__init__.py

class DataPipeline(NamedTuple):
    datasets: torchdata.nodes.BaseNode
    dataloader: torchdata.nodes.BaseNode | torchdata.nodes.Loader

EvalDataPipeline

Bases: NamedTuple

EvalDataPipeline(datasets, dataloader, eval_freq, eval_iterations)

Parameters:

Name	Type	Description	Default
datasets	BaseNode		None
dataloader	BaseNode | Loader		None
eval_freq	int | None		None
eval_iterations	int | None		None

Source code in optimus_dl/modules/data/__init__.py

class EvalDataPipeline(NamedTuple):
    datasets: torchdata.nodes.BaseNode
    dataloader: torchdata.nodes.BaseNode | torchdata.nodes.Loader
    eval_freq: int | None
    eval_iterations: int | None

Modules and Sub-packages

• config:
• datasets:
• presets:
• profiling: Context manager to set the active profiler for the current context.
• transforms: