From 2cfa5719325e943475d3f868370b4f8a4d8258a6 Mon Sep 17 00:00:00 2001 From: takuseno Date: Mon, 6 May 2024 18:07:20 +0900 Subject: [PATCH] Fix default buffer creation at fit_online --- d3rlpy/algos/qlearning/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/d3rlpy/algos/qlearning/base.py b/d3rlpy/algos/qlearning/base.py index a9437114..b1d8791f 100644 --- a/d3rlpy/algos/qlearning/base.py +++ b/d3rlpy/algos/qlearning/base.py @@ -654,7 +654,7 @@ def fit_online( # create default replay buffer if buffer is None: - buffer = create_fifo_replay_buffer(1000000) + buffer = create_fifo_replay_buffer(1000000, env=env) # check action-space assert_action_space_with_env(self, env)