eperezs commited on
Commit
a495d26
·
verified ·
1 Parent(s): 5631118

Update README.md

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