| |
| """Export embeddings from a trained checkpoint.""" |
|
|
| import argparse |
| import sys |
| from pathlib import Path |
|
|
| ROOT = Path(__file__).resolve().parents[1] |
| sys.path.insert(0, str(ROOT)) |
| from src.models.exporting import export_all |
|
|
|
|
| def main() -> None: |
| parser = argparse.ArgumentParser() |
| parser.add_argument("--project-dir", type=Path, required=True) |
| parser.add_argument("--checkpoint", type=Path, required=True) |
| parser.add_argument("--out-dir", type=Path, required=True) |
| parser.add_argument("--chunk", type=int, default=50000) |
| args = parser.parse_args() |
| export_all(args.project_dir, args.checkpoint, args.out_dir, chunk=args.chunk) |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
