amend

pytorch · Apr 8, 2024 · 9a1f5ef · 9a1f5ef
1 parent d02d1e3
commit 9a1f5ef
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sota-implementations/dreamer/dreamer.py b/sota-implementations/dreamer/dreamer.py
@@ -79,9 +79,9 @@ def main(cfg: "DictConfig"):  # noqa: F821
     )
     value_loss = DreamerValueLoss(value_model, discount_loss=True)
 
-    world_model_loss = torch.compile(world_model_loss)
-    actor_loss = torch.compile(actor_loss)
-    value_loss = torch.compile(value_loss)
+    # world_model_loss = torch.compile(world_model_loss)
+    # actor_loss = torch.compile(actor_loss)
+    # value_loss = torch.compile(value_loss)
 
     # Make collector
     collector = make_collector(cfg, train_env, policy)