Update README.md
Browse files
README.md
CHANGED
|
@@ -22,7 +22,7 @@ CLIP_TROHN-Img_Detector is a model presented in the [BiVLC](https://github.com/I
|
|
| 22 |
* Optimizer: Adam optimizer with beta1 = 0.9, beta2 = 0.999, eps = 1e-08 and without weight decay.
|
| 23 |
* Loss function: Binary cross-entropy loss (BCELoss).
|
| 24 |
* Batch size: We define a batch size of 400.
|
| 25 |
-
* Epochs: We trained the text detector over 10 epochs and the image
|
| 26 |
* Data: Then sigmoid neuron is trained with [TROHN-Img](https://huggingface.co/datasets/imirandam/TROHN-Img) dataset.
|
| 27 |
|
| 28 |
### Licensing Information
|
|
|
|
| 22 |
* Optimizer: Adam optimizer with beta1 = 0.9, beta2 = 0.999, eps = 1e-08 and without weight decay.
|
| 23 |
* Loss function: Binary cross-entropy loss (BCELoss).
|
| 24 |
* Batch size: We define a batch size of 400.
|
| 25 |
+
* Epochs: We trained the text detector over 10 epochs and the image detector over 1 epoch. We used validation accuracy as the model selection criterion, i.e. we selected the model with highest accuracy in the corresponding validation set.
|
| 26 |
* Data: Then sigmoid neuron is trained with [TROHN-Img](https://huggingface.co/datasets/imirandam/TROHN-Img) dataset.
|
| 27 |
|
| 28 |
### Licensing Information
|