Skip to content

Commit

Permalink
fix the position of row and col embedding
Browse files Browse the repository at this point in the history
  • Loading branch information
Leaveson committed Dec 24, 2024
1 parent fc0d87b commit c9e5d56
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion rl4co/models/zoo/matnet/decoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class PrecomputedCache:

class MatNetDecoder(AttentionModelDecoder):
def _precompute_cache(self, embeddings: Tuple[Tensor, Tensor], *args, **kwargs):
col_emb, row_emb = embeddings
row_emb, col_emb = embeddings
(
glimpse_key_fixed,
glimpse_val_fixed,
Expand Down

0 comments on commit c9e5d56

Please sign in to comment.