Update

[ghstack-poisoned]
pytorch · Dec 16, 2024 · 1402fae · 1402fae
1 parent c17fea0
commit 1402fae
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sota-implementations/td3/utils.py b/sota-implementations/td3/utils.py
@@ -140,6 +140,8 @@ def make_replay_buffer(
     prefetch: int = 3,
     compile: bool = False,
 ):
+    if compile:
+        prefetch = 0
     with (
         tempfile.TemporaryDirectory()
         if scratch_dir is None