Open vahidsj opened 3 years ago
When running the Infer_example.py I got this error:
Infer_example.py
Traceback (most recent call last): File "infer_example.py", line 162, in <module> inf = Inferer(opt) File "infer_example.py", line 27, in __init__ self.model = opt.model_class(bert, opt).to(opt.device) File "/ukp-storage-1/sadiri/ABSA-PyTorch/models/lcf_bert.py", line 40, in __init__ self.dropout = nn.Dropout(opt.dropout)
Solution: You should add this: opt.dropout = 0.1 in Infer_example.py after setting your trained model, i.e.:
opt.dropout = 0.1
class Option(object): pass opt = Option() opt.model_name = 'lcf_bert' opt.model_class = model_classes[opt.model_name] opt.dataset = 'laptop' opt.dataset_file = dataset_files[opt.dataset] opt.inputs_cols = input_colses[opt.model_name] # set your trained models here opt.state_dict_path = 'state_dict/lcf_bert_laptop_val_acc_0.7806' **opt.dropout = 0.1** opt.embed_dim = 300 opt.hidden_dim = 300 opt.max_seq_len = 85 opt.bert_dim = 768
BTW, if anyone demands, PyABSA provides the training & inference scripts for LCF-BERT as well as other BERT-based and GloVe-based models.
When running the
Infer_example.py
I got this error:Solution: You should add this:
opt.dropout = 0.1
inInfer_example.py
after setting your trained model, i.e.: