How to use Custom Model in MultiAgent PPO Policy

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