amend

pytorch · vmoens · Oct 24, 2023 · Oct 6, 2023 · Oct 6, 2023 · Oct 6, 2023
commit 1fe2cd3e0037df3aefbb22d9304aeda3dd12feec
diff --git a/examples/decision_transformer/utils.py b/examples/decision_transformer/utils.py
@@ -430,6 +430,7 @@ def make_odt_loss(loss_cfg, actor_network):
  alpha_init=loss_cfg.alpha_init,
  target_entropy=loss_cfg.target_entropy,
  )
+ loss.set_keys(action_target="action_cat")
  return loss