amend

pytorch · Jun 19, 2024 · 93ee894 · 93ee894
1 parent faacae4
commit 93ee894
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/sota-implementations/cql/cql_offline.py b/sota-implementations/cql/cql_offline.py
@@ -162,6 +162,7 @@ def main(cfg: "DictConfig"):  # noqa: F821
 
     pbar.close()
     torchrl_logger.info(f"Training time: {time.time() - start_time}")
+    eval_env.close()
 
 
 if __name__ == "__main__":

diff --git a/sota-implementations/cql/cql_online.py b/sota-implementations/cql/cql_online.py
@@ -227,6 +227,8 @@ def main(cfg: "DictConfig"):  # noqa: F821
     torchrl_logger.info(f"Training took {execution_time:.2f} seconds to finish")
 
     collector.shutdown()
+    eval_env.close()
+    train_env.close()
 
 
 if __name__ == "__main__":