Update

[ghstack-poisoned]
pytorch · Dec 16, 2024 · 8206ec0 · 8206ec0
1 parent 63a597b
commit 8206ec0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sota-implementations/td3_bc/utils.py b/sota-implementations/td3_bc/utils.py
@@ -96,7 +96,8 @@ def make_offline_replay_buffer(rb_cfg, device):
         dataset_id=rb_cfg.dataset,
         split_trajs=False,
         batch_size=rb_cfg.batch_size,
-        sampler=SamplerWithoutReplacement(drop_last=False),
+        # drop_last for compile
+        sampler=SamplerWithoutReplacement(drop_last=True),
         prefetch=4,
         direct_download=True,
     )