Skip to content

Commit f5b2c12

Browse files
committed
Also tune the fragment length.
1 parent 45248e6 commit f5b2c12

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/imitation/scripts/config/tuning.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,7 @@ def pc():
201201
"active_selection_oversampling": tune.randint(1, 11),
202202
"comparison_queue_size": tune.randint(1, 1001), # upper bound determined by total_comparisons=1000
203203
"exploration_frac": tune.uniform(0.0, 0.5),
204-
"fragment_length": 100, # TODO: tune this too!
204+
"fragment_length": tune.randint(1, 1001), # trajectories are 1000 steps long
205205
"gatherer_kwargs": {
206206
"temperature": tune.uniform(0.0, 2.0),
207207
"discount_factor": tune.uniform(0.95, 1.0),

0 commit comments

Comments
 (0)