Update README.md
Browse files
README.md
CHANGED
@@ -9,7 +9,7 @@ MediFlow se trata de un modelo inicializado con [xlnet-large-unsased](https://hu
|
|
9 |
para poder realizar Derivaciones Automatizadas en Servicios Hospitalarios. El dataset se puede encontrar de manera pública y se trata de [MedDialog EN](https://arxiv.org/abs/2004.03329).
|
10 |
|
11 |
Para el entrenamiento de este modelo hemos seguidos los estándares de la librería [transformers](https://github.com/huggingface/transformers) y se ha utilizado una
|
12 |
-
NVIDIA P100. Además, ha sido entrenado con un batch-size de 4, un learning rate de
|
13 |
|
14 |
## **Utilización**
|
15 |
Mediante el `pipeline` de Hugging Face:
|
|
|
9 |
para poder realizar Derivaciones Automatizadas en Servicios Hospitalarios. El dataset se puede encontrar de manera pública y se trata de [MedDialog EN](https://arxiv.org/abs/2004.03329).
|
10 |
|
11 |
Para el entrenamiento de este modelo hemos seguidos los estándares de la librería [transformers](https://github.com/huggingface/transformers) y se ha utilizado una
|
12 |
+
NVIDIA P100. Además, ha sido entrenado con un batch-size de 4, un learning rate de 2e-5, X epochs y un weigth decay de 0.015, loggeando los resultados cada 100 iteraciones.
|
13 |
|
14 |
## **Utilización**
|
15 |
Mediante el `pipeline` de Hugging Face:
|