Update rc_position_embedding.py

Thanks @King-Hell for pointing out a bug. We now fix it.
Kris-CN · May 30, 2022 · 7249e68 · 7249e68
1 parent b8f9bb5
commit 7249e68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/PaddleFSL/paddlefsl/backbones/rc_position_embedding.py b/PaddleFSL/paddlefsl/backbones/rc_position_embedding.py
@@ -66,7 +66,7 @@ def forward(self, rc_vector):
  if rc_vector.shape[-1] != self.embedding_dim + 2:
  raise ValueError('Embedding dimension not match. Please check whether'
  'embedding_dimension + 2 == rc_vector.shape[-1]')
- head_position_vector = paddle.to_tensor(rc_vector[:, :, -1], dtype='int64')
+ head_position_vector = paddle.to_tensor(rc_vector[:, :, -2], dtype='int64')
  head_embedding = self.head_position_embedding(head_position_vector)
  tail_position_vector = paddle.to_tensor(rc_vector[:, :, -1], dtype='int64')
  tail_embedding = self.head_position_embedding(tail_position_vector)