update example script

pytorch · Jul 9, 2024 · acc9baf · acc9baf
1 parent 0355caa
commit acc9baf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sota-implementations/td3_bc/td3_bc.py b/sota-implementations/td3_bc/td3_bc.py
@@ -103,7 +103,7 @@ def main(cfg: "DictConfig"):  # noqa: F821
             sampled_tensordict = sampled_tensordict.clone()
 
         # Compute loss
-        q_loss, *_ = loss_module.value_loss(sampled_tensordict)
+        q_loss, *_ = loss_module.qvalue_loss(sampled_tensordict)
 
         # Update critic
         optimizer_critic.zero_grad()