tangxyw / RecAlgorithm

主流推荐系统Rank算法的实现
BSD 2-Clause "Simplified" License
182 stars 46 forks source link

BST error #3

Open linWujl opened 6 months ago

linWujl commented 6 months ago

跑bst的时候训练到中途报如下错误,请问作者知道什么问题吗

tensorflow.python.framework.errors_impl.InvalidArgumentError: 4 root error(s) found. (0) Invalid argument: Incompatible shapes: [102,1,1] vs. [1,3,102,102] [[node model/bst/add_2 (defined at data/code/aftonwu/contUndSta/dev/transformer_layer.py:124) ]] [[replica_1/model/bst_1/ExpandDims/_413]] (1) Invalid argument: Incompatible shapes: [102,1,1] vs. [1,3,102,102] [[node model/bst/add_2 (defined at data/code/aftonwu/contUndSta/dev/transformer_layer.py:124) ]] [[Adam/concat_18/_1394]] (2) Invalid argument: Incompatible shapes: [102,1,1] vs. [1,3,102,102] [[node model/bst/add_2 (defined at data/code/aftonwu/contUndSta/dev/transformer_layer.py:124) ]] [[add_4/ReadVariableOp_1/_996]] (3) Invalid argument: Incompatible shapes: [102,1,1] vs. [1,3,102,102] [[node model/bst/add_2 (defined at data/code/aftonwu/contUndSta/dev/transformer_layer.py:124) ]] 0 successful operations. 0 derived errors ignored. [Op:__inference_train_function_78818]

Errors may have originated from an input operation. Input Source operations connected to node model/bst/add_2: model/bst/ExpandDims_2 (defined at data/code/aftonwu/contUndSta/dev/transformer_layer.py:119) model/bst/truediv (defined at data/code/aftonwu/contUndSta/dev/transformer_layer.py:123)