rl4co
rl4co copied to clipboard
Fixed some parameter discrepancy for MDAM
Fixed some parameter discrepancy for MDAM.
But the encoder still raises ValueError.
See GraphAttentionEncoder
at models/zoo/mdam/encoder.py
.
ValueError: too many values to unpack (expected 3); here self.attention_layer(h_old) is a single tensor
h_new, attn, V = self.attention_layer(h_old)