Skip to content

Commit b20b2db

Browse files
committed
Make PCL check n_episodes before calling sample_episodes
1 parent abc9165 commit b20b2db

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

chainerrl/agents/pcl.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -273,6 +273,9 @@ def update_from_replay(self):
273273
if len(self.replay_buffer) < self.replay_start_size:
274274
return
275275

276+
if self.replay_buffer.n_episodes < self.batchsize:
277+
return
278+
276279
if self.process_idx == 0:
277280
self.logger.debug('update_from_replay')
278281

0 commit comments

Comments
 (0)