From c9e5d56c911e3c4ad8dec6a58be197ff170c4d9e Mon Sep 17 00:00:00 2001 From: Leaveson Date: Tue, 24 Dec 2024 07:54:46 +0000 Subject: [PATCH] fix the position of row and col embedding --- rl4co/models/zoo/matnet/decoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rl4co/models/zoo/matnet/decoder.py b/rl4co/models/zoo/matnet/decoder.py index 5a8d6e28..37417968 100644 --- a/rl4co/models/zoo/matnet/decoder.py +++ b/rl4co/models/zoo/matnet/decoder.py @@ -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,