DTLoss

class torchrl.objectives.DTLoss(*args, **kwargs)

TorchRL 对在线决策 Transformer 损失的实现。

在 “Decision Transformer: Reinforcement Learning via Sequence Modeling” <https://arxiv.org/abs/2106.01345> 中提出

参数:

actor_network (ProbabilisticActor) – 随机策略

关键字参数:

• loss_function (str) – 要使用的损失函数。默认为 "l2"。

• reduction (str, optional) – 指定应用于输出的约简："none" | "mean" | "sum"。"none"：不应用约简，"mean"：输出的总和将除以输出中的元素数量，"sum"：将对输出进行求和。默认为 "mean"。

default_keys

别名：_AcceptedKeys

forward(tensordict: TensorDictBase = None) → TensorDictBase

计算在线决策 Transformer 的损失。