error when training an extractive summarization model

HHousen / TransformerSum

Models to perform neural summarization (extractive and abstractive) using machine learning transformers and a tool to convert abstractive summarization datasets to the extractive task.

https://transformersum.rtfd.io

GNU General Public License v3.0

428 stars 58 forks source link

error when training an extractive summarization model #63

Closed CoderBinGe closed 2 years ago

CoderBinGe commented 3 years ago

I have tried O1, O2, etc… but still reported an error. How to set "amp_level" parameter? When don't support with amp_backend='apex'.

Sere-Fu commented 3 years ago

I have tried O1, O2, etc… but still reported an error. How to set "amp_level" parameter? When don't support with amp_backend='apex'.

Simply remove the specific arg.

CoderBinGe commented 2 years ago

ok, Thanks!