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)