From 684027ac55362109eec2d81736d1a423f8e8658e Mon Sep 17 00:00:00 2001
From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com>
Date: Fri, 25 Oct 2024 23:51:08 -0400
Subject: [PATCH] reset torch compiler cache just in case

---
 pokemonred_puffer/sweep.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pokemonred_puffer/sweep.py b/pokemonred_puffer/sweep.py
index 326bd13..e4820a9 100644
--- a/pokemonred_puffer/sweep.py
+++ b/pokemonred_puffer/sweep.py
@@ -179,6 +179,10 @@ def launch_agent(
     debug: bool = False,
 ):
     def _fn():
+        import torch
+
+        torch.compiler.reset()
+
         agent_config: DictConfig = OmegaConf.load(os.environ["WANDB_SWEEP_PARAM_PATH"]).x.value
         agent_config = update_base_config(base_config, agent_config)
         train.train(config=agent_config, debug=debug, track=True)