Thanks for your reply, I solved my problem by copying the code of GTRXL from attention_net.py to custom_model and modifying it.
Thanks for your reply, I solved my problem by copying the code of GTRXL from attention_net.py to custom_model and modifying it.