enc_output,enc_slf_attn = self.slf_attn(user_embedding,item_input,item_input,mask = slf_attn_mask)
实际上只有一个返回值,但是却写了两个返回值,所以报错。
改正为
enc_output = self.slf_attn(user_embedding,item_input,item_input,mask = slf_attn_mask)
enc_output,enc_slf_attn = self.slf_attn(user_embedding,item_input,item_input,mask = slf_attn_mask)
实际上只有一个返回值,但是却写了两个返回值,所以报错。
改正为
enc_output = self.slf_attn(user_embedding,item_input,item_input,mask = slf_attn_mask)