Skip to content
PyLate
get_rank
Initializing search
lighton/pylate
Documentation
Models
API reference
PyLate
lighton/pylate
Documentation
Models
API reference
API reference
Overview
evaluation
hf_hub
indexes
losses
models
rank
retrieve
scores
utils
utils
ColBERTCollator
KDProcessing
all_gather_with_gradients
all_gather
convert_to_tensor
get_rank
get_world_size
iter_batch
get_rank
¶
Returns the current rank in a distributed training.