Update README.md
Browse files
README.md
CHANGED
|
@@ -54,7 +54,7 @@ for text in texts:
|
|
| 54 |
scores[text] = outputs.logits[0,1].item()
|
| 55 |
```
|
| 56 |
|
| 57 |
-
Here is how to use this model to
|
| 58 |
|
| 59 |
```python
|
| 60 |
from transformers import BridgeTowerProcessor, BridgeTowerForMaskedLM
|
|
@@ -99,7 +99,7 @@ The model was pre-trained for 10 epochs on an Intel AI supercomputing cluster us
|
|
| 99 |
The optimizer used was AdamW with a learning rate of 1e-7. No data augmentation was used except for center-crop. The image resolution in pre-training is set to 294 x 294.
|
| 100 |
|
| 101 |
## Evaluation results
|
| 102 |
-
Please refer to [Table 5](https://arxiv.org/pdf/2206.08657.pdf) for BridgeTower's performance on Image Retrieval and other
|
| 103 |
|
| 104 |
### BibTeX entry and citation info
|
| 105 |
```bibtex
|
|
|
|
| 54 |
scores[text] = outputs.logits[0,1].item()
|
| 55 |
```
|
| 56 |
|
| 57 |
+
Here is how to use this model to perform masked language modeling:
|
| 58 |
|
| 59 |
```python
|
| 60 |
from transformers import BridgeTowerProcessor, BridgeTowerForMaskedLM
|
|
|
|
| 99 |
The optimizer used was AdamW with a learning rate of 1e-7. No data augmentation was used except for center-crop. The image resolution in pre-training is set to 294 x 294.
|
| 100 |
|
| 101 |
## Evaluation results
|
| 102 |
+
Please refer to [Table 5](https://arxiv.org/pdf/2206.08657.pdf) for BridgeTower's performance on Image Retrieval and other downstream tasks.
|
| 103 |
|
| 104 |
### BibTeX entry and citation info
|
| 105 |
```bibtex
|