How to get more info data in reward model? about di-engine

[xuehui1991]

@zjowowen

[PaParaZz1]

In most of DI-engine's policies, we use process_transition method to pass necessary information (e.g. link), so we don't add other attributes of env timestep like info into transition.

If you want to add this, briefly, there are two methods:

• modify process_transition method of policy you used, add info into transition, and it will be also available in reward_model
• add the information you want into obs or reward field, and process them in reward_model

[xuehui1991]

Thank you and I already solve this problem!