|
--- |
|
language: |
|
- en |
|
license: apache-2.0 |
|
tags: |
|
- dialogue policy |
|
- task-oriented dialog |
|
datasets: |
|
- ConvLab/multiwoz21 |
|
|
|
--- |
|
|
|
# mle-policy-multiwoz21 |
|
|
|
This is a MLE model trained on [MultiWOZ 2.1](https://huggingface.co/datasets/ConvLab/multiwoz21) |
|
|
|
Refer to [ConvLab-3](https://github.com/ConvLab/ConvLab-3) for model description and usage. |
|
|
|
## Training procedure |
|
|
|
### Training hyperparameters |
|
|
|
The following hyperparameters were used during training: |
|
- learning_rate: 0.0001 |
|
- train_batch_size: 32 |
|
- seed: 0 |
|
- optimizer: Adam |
|
- num_epochs: 24 |
|
- use checkpoint which performed best on validation set |
|
### Framework versions |
|
- Transformers 4.18.0 |
|
- Pytorch 1.10.2+cu111 |
|
|
