During reproduction on MELD-DA, I traced the NaN source to trans_encoder
in the concept branch (HIER.py). Bypassing it made the pipeline stable.
Could you clarify:
-
Was trans_encoder numerically stable in your original setup, and was
there any specific initialization, dtype handling, or stabilization applied
that is not reflected in the released code?
-
If trans_encoder is bypassed, how much would you expect it to affect
the final performance?
During reproduction on MELD-DA, I traced the NaN source to
trans_encoderin the concept branch (
HIER.py). Bypassing it made the pipeline stable.Could you clarify:
Was
trans_encodernumerically stable in your original setup, and wasthere any specific initialization, dtype handling, or stabilization applied
that is not reflected in the released code?
If
trans_encoderis bypassed, how much would you expect it to affectthe final performance?