search

taoqi98 / FedNewsRec

Data and code of our paper "Privacy-Preserving News Recommendation Model Learning"
MIT License
39 stars 14 forks source link

InvalidArgumentError #4

Closed cqliang1995 closed 3 years ago

cqliang1995 commented 3 years ago

when I run your code，I got that：

tensorflow.python.framework.errors_impl.InvalidArgumentError: Dimension must be 5 but is 4 for 'attention_1/transpose_7' (op: 'Transpose') with input shapes: [?,20,?,20,?], [4].

taoqi98 commented 3 years ago

This is caused by the unmatched versions of tensor-flow and Keras. You can use tensorflow-gpu == 1.13.0 and keras == 2.2.4 to run our codes.