From bfc891f10ff5e2883c0895a8acd99c363ba3ec6b Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Sat, 14 Dec 2024 19:19:13 -0800 Subject: [PATCH] Update (base update) [ghstack-poisoned] --- sota-implementations/gail/gail.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sota-implementations/gail/gail.py b/sota-implementations/gail/gail.py index 48e3b18c2c3..072063bb689 100644 --- a/sota-implementations/gail/gail.py +++ b/sota-implementations/gail/gail.py @@ -93,6 +93,7 @@ def main(cfg: "DictConfig"): # noqa: F821 ), sampler=SamplerWithoutReplacement(), batch_size=cfg.ppo.loss.mini_batch_size, + compilable=cfg.compile.compile, ) # Create loss and adv modules