jacobin-sctcs / nltodb Goto Github PK

File "D:\NLTODB-master\train.py", line 43, in
train_model(model,n_epochs,optimizer,train_dataloader,valid_dataloader,train_entry)
File "D:\NLTODB-master\utils.py", line 130, in train_model
scores = model(data['question_tokens'] , data['column_headers'] ,train_entry , data['where_col'] ,
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\NLTODB-master\cuda\Lib\site-packages\torch\nn\modules\module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\NLTODB-master\cuda\Lib\site-packages\torch\nn\modules\module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\NLTODB-master\model.py", line 44, in forward
embedding , length = self.wordembedding.gen_x_batch(question,columns)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\NLTODB-master\wordembedding.py", line 49, in gen_x_batch
val_emb_array[i,j,:] = val_embs[i][j]
~~~~~~~~~~~~~^^^^^^^
ValueError: could not broadcast input array from shape (300,) into shape (50,)