smallsmallwood

I follow your requirements ,and when running"python demo.py final", i get the erors , and i have run setup_all.sh file before running demo.py

1

Traceback (most recent call last): File "demo.py", line 14, in import coref_model as cm File "/home/wlk/PycharmProjects/e2e/coref_model.py", line 10, in import coref_ops File "/home/wlk/PycharmProjects/e2e/coref_ops.py", line 4, in coref_op_library = tf.load_op_library("./coref_kernels.so") File...

Code is right ???

1

**orginal code:** ``` def forward(self, document_batch: torch.Tensor, device='cpu', bert_batch_size=0): **bert_output = torch.zeros(size=(document_batch.shape[0], min(document_batch.shape[1], bert_batch_size), self.bert.config.hidden_size), dtype=torch.float, device=device) for doc_id in range(document_batch.shape[0]): bert_output[doc_id][:bert_batch_size] = self.dropout(self.bert(document_batch[doc_id][:bert_batch_size,0], token_type_ids=document_batch[doc_id][:bert_batch_size, 1], attention_mask=document_batch[doc_id][:bert_batch_size, 2])[1])** output, (_,...