imirandam commited on
Commit
02f3011
·
verified ·
1 Parent(s): 5c3561e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 detectors 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
 
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