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
|