paq/rerankers/rerank.py [67:75]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    if cuda:
        model = model.cuda()
        model = to_fp16(model) if fp16 else model

    t = time.time()

    def log_progress(j, outputs):
        t2 = time.time()
        logger.info(
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



paq/retrievers/embed.py [33:41]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    if cuda:
        model = model.cuda()
        model = to_fp16(model) if fp16 else model

    t = time.time()

    def log_progress(j, outputs):
        t2 = time.time()
        logger.info(
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -