File size: 275 Bytes
edcf5ee | 1 2 3 4 5 6 7 8 | from .data_simmim import build_loader_simmim
from .data_finetune import build_loader_finetune
def build_loader(config, logger, is_pretrain):
if is_pretrain:
return build_loader_simmim(config, logger)
else:
return build_loader_finetune(config, logger) |