mirror of
https://github.com/wassname/ray.git
synced 2026-08-13 12:30:18 +08:00
[rllib] Rename sample_batch_size => rollout_fragment_length (#7503)
* bulk rename * deprecation warn * update doc * update fig * line length * rename * make pytest comptaible * fix test * fi sys * rename * wip * fix more * lint * update svg * comments * lint * fix use of batch steps
This commit is contained in:
@@ -22,10 +22,11 @@ class AsyncSamplesOptimizer(PolicyOptimizer):
|
||||
This class coordinates the data transfers between the learner thread
|
||||
and remote workers (IMPALA actors).
|
||||
"""
|
||||
|
||||
def __init__(self,
|
||||
workers,
|
||||
train_batch_size=500,
|
||||
sample_batch_size=50,
|
||||
rollout_fragment_length=50,
|
||||
num_envs_per_worker=1,
|
||||
num_gpus=0,
|
||||
lr=0.0005,
|
||||
@@ -90,7 +91,7 @@ class AsyncSamplesOptimizer(PolicyOptimizer):
|
||||
max_sample_requests_in_flight_per_worker),
|
||||
replay_buffer_num_slots=replay_buffer_num_slots,
|
||||
train_batch_size=train_batch_size,
|
||||
sample_batch_size=sample_batch_size,
|
||||
rollout_fragment_length=rollout_fragment_length,
|
||||
broadcast_interval=broadcast_interval)
|
||||
else:
|
||||
self.aggregator = SimpleAggregator(
|
||||
@@ -100,7 +101,7 @@ class AsyncSamplesOptimizer(PolicyOptimizer):
|
||||
max_sample_requests_in_flight_per_worker),
|
||||
replay_buffer_num_slots=replay_buffer_num_slots,
|
||||
train_batch_size=train_batch_size,
|
||||
sample_batch_size=sample_batch_size,
|
||||
rollout_fragment_length=rollout_fragment_length,
|
||||
broadcast_interval=broadcast_interval)
|
||||
|
||||
def add_stat_val(self, key, val):
|
||||
|
||||
Reference in New Issue
Block a user