lsdefine / attention-is-all-you-need-keras

A Keras+TensorFlow Implementation of the Transformer: Attention Is All You Need
702 stars 188 forks source link

Keras and Tensorflow Versions #9

Open amirveyseh opened 5 years ago

amirveyseh commented 5 years ago

When I run the code I got this error on line 100 of transformer.py:

ValueError: Axis 0 of input tensor should have a defined dimension, but is None. Full tensor shape: (None, None, None). Typically you need to pass a fully-defined input_shape argument to your first layer.

could you specify the versions of keras and tensorflow that you used for your test?

lsdefine commented 5 years ago

keras 2.1.3, tensorflow 1.4.1