@sarankarthik you just need to add it to your config.yml file and then it will also train the custom policy
config.yml