From 95ffff36596095d091e79b0eb43fb67eb721b8ea Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Fri, 13 Dec 2024 14:13:58 -0800 Subject: [PATCH] Update [ghstack-poisoned] --- sota-implementations/discrete_sac/discrete_sac.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sota-implementations/discrete_sac/discrete_sac.py b/sota-implementations/discrete_sac/discrete_sac.py index 1fa3271da82..8b3efe15102 100644 --- a/sota-implementations/discrete_sac/discrete_sac.py +++ b/sota-implementations/discrete_sac/discrete_sac.py @@ -180,7 +180,7 @@ def update(sampled_tensordict): with timeit("update"): torch.compiler.cudagraph_mark_step_begin() sampled_tensordict = sampled_tensordict.to(device) - loss_out = update(sampled_tensordict) + loss_out = update(sampled_tensordict).clone() tds.append(loss_out)