From 8206ec03136c4534e3fcb2138b364e2326c30d2b Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Sun, 15 Dec 2024 20:06:43 -0800 Subject: [PATCH] Update [ghstack-poisoned] --- sota-implementations/td3_bc/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sota-implementations/td3_bc/utils.py b/sota-implementations/td3_bc/utils.py index 7e17a8206ef..c7b99e4f0e3 100644 --- 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, )