执行ELMO模型时出错-tensorflow.python.framework.errors_impl.InvalidArgumentError:

时间:2019-01-02 16:00:58

标签: tensorflow keras word-embedding

我尝试运行ELMO模型,并遵循this示例。但是,我遇到错误了

File "ELMOS2.py", line 225, in <module> validation_steps=valid_steps) File "/home/ec2-user/.local/lib/python3.6/site-packages/keras/legacy/interfaces.py", line 91, in wrapper return func(*args, **kwargs) File "/home/ec2-user/.local/lib/python3.6/site-packages/keras/engine/training.py", line 1418, in fit_generator initial_epoch=initial_epoch) File "/home/ec2-user/.local/lib/python3.6/site-packages/keras/engine/training_generator.py", line 217, in fit_generator class_weight=class_weight) File "/home/ec2-user/.local/lib/python3.6/site-packages/keras/engine/training.py", line 1217, in train_on_batch outputs = self.train_function(ins) File "/home/ec2-user/.local/lib/python3.6/site-packages/keras/backend/tensorflow_backend.py", line 2715, in __call__ return self._call(inputs) File "/home/ec2-user/.local/lib/python3.6/site-packages/keras/backend/tensorflow_backend.py", line 2675, in _call fetched = self._callable_fn(*array_vals) File "/home/ec2-user/.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1439, in __call__ run_metadata_ptr) File "/home/ec2-user/.local/lib/python3.6/site-packages/tensorflow/python/framework/errors_impl.py", line 528, in __exit__ c_api.TF_GetCode(self.status.status)) tensorflow.python.framework.errors_impl.InvalidArgumentError: indices[6,69] = 280617 is not in [0, 280617) [[{{node embedding_1/embedding_lookup}} = GatherV2[Taxis=DT_INT32, Tindices=DT_INT32, Tparams=DT_FLOAT, _class=["loc:@training/Adam/Assign_2"], _device="/job:localhost/replica:0/task:0/device:CPU:0"](embedding_1/embeddings/read, _arg_input_1_0_0, training/Adam/gradients/embedding_1/embedding_lookup_grad/concat/axis)]]

0 个答案:

没有答案