Python EpochGen Examples

Example #1

def get_loader(data, args):
    data = EpochGen(data, batch_size=args.batch_size, shuffle=False)
    return data

Example #2

File: train.py Project: strategist922/MSMARCOV2

def get_loader(data, config):
    data = EpochGen(data,
                    batch_size=config.get('training',
                                          {}).get('batch_size', 32),
                    shuffle=True)
    return data

Example #3

File: predict.py Project: yongbowin/MSMARCO-Question-Answering_annotation

def get_loader(data, args):
    data = EpochGen(
        data,
        batch_size=args.batch_size,  # predict batch size, default=64
        shuffle=False)
    return data