torch==1.10.0 tensorboardX==2.4.1 pyrouge==0.1.3 pytorch_pretrained_bert==0.6.2 onnx-simplifier==0.3.6