Mirrors/the-algorithm
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm.git synced 2024-06-13 06:38:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update faiss_index_bq_dataset.py

Browse Source 
torch & tensorflow pull request
This commit is contained in:
ii_ruatnec 2023-04-05 18:41:02 +09:00 committed by GitHub
parent 138bb51997
commit 528f94b2cc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ann/src/main/python/dataflow/faiss_index_bq_dataset.py
View File

@ -4,6 +4,8 @@ import os
import pkgutil
import sys
from urllib.parse import urlsplit
import torch as nn
import tensorflow as tf
import apache_beam as beam
from apache_beam.options.pipeline_options import PipelineOptions