primeqa.ir.dense.dpr_top.dpr.config.DPRIndexingArguments

class primeqa.ir.dense.dpr_top.dpr.config.DPRIndexingArguments(output_dir: str, bsize: int = 16, collection: str = 'None', ctx_encoder_name_or_path: str = 'facebook/dpr-ctx_encoder-multiset-base', embed: str = '1of1', sharded_index: bool = True)

Bases: object

Arguments used in indexing

Methods

Attributes

bsize
collection
ctx_encoder_name_or_path
embed
sharded_index
output_dir