Skip to content

Commit

Permalink
Merge pull request #235 from ai4co/fix/matnet
Browse files Browse the repository at this point in the history
[Fix] Fix the position of row and col embedding
  • Loading branch information
fedebotu authored Dec 26, 2024
2 parents fc0d87b + c9e5d56 commit 2d916ba
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 2d916ba

Please sign in to comment.