1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 03:30:19 +01:00

waiting for a new checkpoint if it's available

This commit is contained in:
Ajay Deshpande
2018-10-05 19:08:24 -07:00
committed by zach dwiel
parent 5eac0102de
commit 7f00235ed5
7 changed files with 49 additions and 20 deletions

View File

@@ -251,6 +251,9 @@ class ClippedPPOAgent(ActorCriticAgent):
# clean memory
self.call_memory('clean')
def _should_train_helper(self, wait_for_full_episode=True):
return super()._should_train_helper(True)
def train(self):
if self._should_train(wait_for_full_episode=True):
for network in self.networks.values():