dpr_scale/run_retrieval_fb.py [14:37]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
try:
    # dummy import to make Manifold paths happy
    from pytext import fb  # noqa
except ImportError:
    print("Failed to import pytext. Ignore this if in OSS.")

import ujson as json


def get_logger():
    log_format = "[%(asctime)s] [%(levelname)s]: %(message)s"
    logging.basicConfig(format=log_format, level=logging.INFO)
    logger = logging.getLogger(__name__)
    return logger


def get_parser():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--ctx_embeddings_dir",
        type=str,
        default="",
    )
    parser.add_argument(
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



dpr_scale/run_retrieval_multiset.py [14:37]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
try:
    # dummy import to make Manifold paths happy
    from pytext import fb  # noqa
except ImportError:
    print("Failed to import pytext. Ignore this if in OSS.")

import ujson as json


def get_logger():
    log_format = "[%(asctime)s] [%(levelname)s]: %(message)s"
    logging.basicConfig(format=log_format, level=logging.INFO)
    logger = logging.getLogger(__name__)
    return logger


def get_parser():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--ctx_embeddings_dir",
        type=str,
        default="",
    )
    parser.add_argument(
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -