MCPcopy Create free account
hub / github.com/dcharatan/flowmap / DataModulePretrainCfg

Class DataModulePretrainCfg

flowmap/dataset/data_module_pretrain.py:26–28  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

24
25@dataclass
26class DataModulePretrainCfg:
27 train: DataLoaderStageCfg
28 val: DataLoaderStageCfg
29
30
31DatasetShim = Callable[[Dataset, Stage], Dataset]

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected