Infersent: 인코더/play.ipynb 실행 중 오류 발생

다음 줄을 실행할 때 :

임베딩 = model.encode(문장, bsize=128, tokenize=False, verbose=True)
print('nb 문장 인코딩: {0}'.format(len(embeddings)))

다음 오류가 발생합니다.

Nb 단어 유지 : 128201/130068 (98.56 %)

RuntimeError Traceback(가장 최근 호출 마지막)
~에()
----> 1개의 임베딩 = model.encode(문장, bsize=128, tokenize=False, verbose=True)
2 print('nb 문장 인코딩: {0}'.format(len(embeddings)))

인코딩의 /home/leena/Downloads/InferSent-master/encoder/models.py(self, sentence, bsize, tokenize, verbose)
177 if self.use_cuda:
178화
--> 179 배치 = self.forward((배치, 길이[stidx:stidx + bsize])).data.cpu().numpy()
180 임베딩.append(배치)
181 임베딩 = np.vstack(임베딩)

/home/leena/Downloads/InferSent-master/encoder/models.py 앞으로(self, sent_tuple)
48
49 # 길이로 정렬 해제
---> 50 idx_unsort = torch.from_numpy(idx_unsort).cuda() if self.use_cuda else torch.from_numpy(idx_sort)
51화
52

런타임 오류: from_numpy는 np.ndarray를 예상하지만 torch.LongTensor를 얻었습니다.

버전 세부정보:
'3.6.0 |아나콘다 커스텀(64비트)| (기본값, 2016년 12월 23일, 12:22:00) \n[GCC 4.4.7 20120313 (Red Hat 4.4.7-1)]'
에 [ ]:
토치-0.1.12.post2