Comments (1)
Few questions:
- Have you tried using the PyTorch data profiler? https://pytorch.org/tutorials/recipes/recipes/profiler_recipe.html
I'd probably start there. - How are you loading the data? It looks like your dataset is loaded from memory, but I want to confirm there's not an issue with the loading step. PyTorch has specific classes:
- Creating a custom dataset: https://pytorch.org/tutorials/beginner/basics/data_tutorial.html#creating-a-custom-dataset-for-your-files
- Creating a data loader for that dataset: https://pytorch.org/tutorials/beginner/basics/data_tutorial.html#preparing-your-data-for-training-with-dataloaders
- What value are you setting for
index_bsize
? You probably want to increase this value until it breaks and then bring it back down.
If data transfers are frequently going back and forth between the CPU and GPU, that will bottleneck a lot of GPU processing.
from colbert.
Related Issues (20)
- What does the nway document actually indicate? HOT 1
- Why is the labels initialised to zeros? HOT 1
- RuntimeError: quantile() input tensor is too large HOT 4
- Set batch size when indexing HOT 3
- Pre-filtering the documents based on metadata before late-interaction HOT 5
- What is Colbert v1.9?
- Issue: Training "resume" and "resume_optimizer" implementation was removed
- Irrelevant results returned by the Colbert V2 Model HOT 1
- crypt.h: No such file or directory HOT 7
- Basic Training (ColBERTv1-style) -> ujson.JSONDecodeError: Expected object or value HOT 2
- How can I use "all_mpnet_base_v2" model for colbert indexing and searching?
- GPU not working while training a new model in Colab
- [rank1]:[E ProcessGroupNCCL.cpp:523] [Rank 1] Watchdog caught collective operation timeout: WorkNCCL HOT 2
- How to set chunk_size
- Tokens in `skiplist` are not returned (masked out) but they still affect other tokens embeddings. Is this expected? HOT 2
- How to get the mapping information about doc_id with doc_content. HOT 1
- CollectionEncoder blocking on encoder N passages HOT 1
- Focusing retrieval on list of document ids with doc_ids parameter doesn't work
- type object 'ColBERT' has no attribute 'segmented_maxsim' HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from colbert.