ddpt-policy-sgd / README.md
ChrisGeishauser's picture
Update README.md
eb56b85
|
raw
history blame
No virus
701 Bytes
metadata
language:
  - en
license: apache-2.0
tags:
  - dialogue policy
  - task-oriented dialog
datasets:
  - ConvLab/sgd

ddpt-policy-sgd

This is a DDPT model (https://aclanthology.org/2022.coling-1.21/) trained on Schema-Guided Dialog

Refer to ConvLab-3 for model description and usage.

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 1e-05
  • train_batch_size: 64
  • seed: 0
  • optimizer: Adam
  • num_epochs: 40.0
  • use checkpoint which performed best on validation set

Framework versions

  • Transformers 4.18.0
  • Pytorch 1.10.2+cu111