UMS for Multi-turn Response Selection in PyTorch
UMS for Multi-turn Response Selection
PyTorch Implementation for AAAI’21 “Do Response Selection Models Really Know What’s Next? Utterance Manipulation Strategies for Multi-turn Response Selection”
Implements the model described in the following paper Do Response Selection Models Really Know What’s Next? Utterance Manipulation Strategies for Multi-turn Response Selection.
@inproceedings{whang2021ums,
title={Do Response Selection Models Really Know What's Next? Utterance Manipulation Strategies for Multi-turn Response Selection},
author={Whang, Taesun and Lee, Dongyub and Oh, Dongsuk and Lee, Chanhee and Han, Kijong and Lee, Dong-hun and Lee, Saebyeok},
booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
year={2021}
}
This code is reimplemented as a fork of huggingface/transformers and taesunwhang/BERT-ResSel.
Setup and Dependencies
This code is implemented using PyTorch v1.6.0, and provides out of the box support with CUDA 10.1 and