Skip to content

Commit 8efd26e

Browse files
committed
changed the n_steps back to the original values
1 parent 92e7f83 commit 8efd26e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

Diff for: reproductions/offline/qdt.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def fit_cql(
148148

149149
cql.fit(
150150
dataset,
151-
n_steps=3000, # 500000,
151+
n_steps=500000,
152152
n_steps_per_epoch=1000,
153153
save_interval=50,
154154
evaluators={"environment": d3rlpy.metrics.EnvironmentEvaluator(env)},
@@ -212,7 +212,7 @@ def callback(algo: d3rlpy.algos.IQL, epoch: int, total_step: int) -> None:
212212

213213
iql.fit(
214214
dataset,
215-
n_steps=3000, # 500000,
215+
n_steps=500000,
216216
n_steps_per_epoch=1000,
217217
save_interval=10,
218218
callback=callback,
@@ -281,7 +281,7 @@ def fit_dt(
281281

282282
dt.fit(
283283
dataset,
284-
n_steps=3000, # 100000,
284+
n_steps=100000,
285285
n_steps_per_epoch=1000,
286286
save_interval=10,
287287
eval_env=env,

0 commit comments

Comments
 (0)