From 8476c5ec870a5961dcccb3922c3beef7c19ff8a6 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Sat, 14 Dec 2024 20:24:02 -0800 Subject: [PATCH] Update [ghstack-poisoned] --- sota-implementations/gail/gail.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sota-implementations/gail/gail.py b/sota-implementations/gail/gail.py index 072063bb689..1075a78eba6 100644 --- a/sota-implementations/gail/gail.py +++ b/sota-implementations/gail/gail.py @@ -233,7 +233,7 @@ def update(data, expert_data, num_network_updates=num_network_updates): # Update the networks optim.step() - return TensorDict(dloss=d_loss, alpha=alpha).detach() + return {"dloss": d_loss, "alpha": alpha} if cfg.compile.compile: update = compile_with_warmup(update, warmup=2, mode=compile_mode)