Thanks for the code, can you please share if there are any visualization tools for visualizing mmTransformer? Regards, Sai