Huggingface DecisionTransformer - Reward Calculation

Hi guys,

I am using Huggingface Decision Transformer Model (DecisionTransformerModel) , I would like to get the predicted action and calculate and update return in each step according to predicted action. When I look to the source code, I couldn’t see where the given rewards are processed. Thanks in advance :slight_smile:

3 Likes