We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
PyTorch Transformer Dialogue Model
trs
How to run:
python train.py -s trs_small --device cuda:0