Unverified Commit 9617b130 authored by Saad Mahmud's avatar Saad Mahmud Committed by GitHub
Browse files

[Doctest] Add configuration_dpr.py (#20080)

* Add example docstring for DPRConfig

* Add DPRConfig to documentation_tests
parent a0f86743
...@@ -92,7 +92,21 @@ class DPRConfig(PretrainedConfig): ...@@ -92,7 +92,21 @@ class DPRConfig(PretrainedConfig):
projection_dim (`int`, *optional*, defaults to 0): projection_dim (`int`, *optional*, defaults to 0):
Dimension of the projection for the context and question encoders. If it is set to zero (default), then no Dimension of the projection for the context and question encoders. If it is set to zero (default), then no
projection is done. projection is done.
"""
Example:
```python
>>> from transformers import DPRConfig, DPRContextEncoder
>>> # Initializing a DPR facebook/dpr-ctx_encoder-single-nq-base style configuration
>>> configuration = DPRConfig()
>>> # Initializing a model (with random weights) from the facebook/dpr-ctx_encoder-single-nq-base style configuration
>>> model = DPRContextEncoder(configuration)
>>> # Accessing the model configuration
>>> configuration = model.config
```"""
model_type = "dpr" model_type = "dpr"
def __init__( def __init__(
......
...@@ -62,6 +62,7 @@ src/transformers/models/deit/modeling_tf_deit.py ...@@ -62,6 +62,7 @@ src/transformers/models/deit/modeling_tf_deit.py
src/transformers/models/detr/configuration_detr.py src/transformers/models/detr/configuration_detr.py
src/transformers/models/detr/modeling_detr.py src/transformers/models/detr/modeling_detr.py
src/transformers/models/distilbert/configuration_distilbert.py src/transformers/models/distilbert/configuration_distilbert.py
src/transformers/models/dpr/configuration_dpr.py
src/transformers/models/dpt/modeling_dpt.py src/transformers/models/dpt/modeling_dpt.py
src/transformers/models/electra/configuration_electra.py src/transformers/models/electra/configuration_electra.py
src/transformers/models/electra/modeling_electra.py src/transformers/models/electra/modeling_electra.py
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment