search

THU-BPM / SelfORE

The source code of paper "SelfORE: Self-supervised Relational Feature Learning for Open Relation Extraction"
47 stars 10 forks source link

Issue in Classifier.py #16

Open sefeoglu opened 1 year ago

sefeoglu commented 1 year ago

Hi, I tried to run the codes, and there is problem at the line 136 in classifier.py. The problem is : 

    total_train_loss += loss.item()
AttributeError: 'str' object has no attribute 'item'

Is it related to version of the transformer?

Thanks.