We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 45248e6 commit f5b2c12Copy full SHA for f5b2c12
src/imitation/scripts/config/tuning.py
@@ -201,7 +201,7 @@ def pc():
201
"active_selection_oversampling": tune.randint(1, 11),
202
"comparison_queue_size": tune.randint(1, 1001), # upper bound determined by total_comparisons=1000
203
"exploration_frac": tune.uniform(0.0, 0.5),
204
- "fragment_length": 100, # TODO: tune this too!
+ "fragment_length": tune.randint(1, 1001), # trajectories are 1000 steps long
205
"gatherer_kwargs": {
206
"temperature": tune.uniform(0.0, 2.0),
207
"discount_factor": tune.uniform(0.95, 1.0),
0 commit comments