ImportError: cannot import name 'amp'

ThilinaRajapakse / pytorch-transformers-classification

Based on the Pytorch-Transformers library by HuggingFace. To be used as a starting point for employing Transformer models in text classification tasks. Contains code to easily train BERT, XLNet, RoBERTa, and XLM models for text classification.

Apache License 2.0

306 stars 97 forks source link

ImportError: cannot import name 'amp' #35

Open vr25 opened 4 years ago

vr25 commented 4 years ago

Hi,

Can you please take a look at this issue?

Thanks!

Jumponthemoon commented 3 years ago

Try upgrade torch to a higher version,torch=1,.6.0 works for me.