gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 11:40:18 +01:00
Code Issues Projects Releases Wiki Activity

Ignoring redis sub if testing

Browse Source
This commit is contained in:
Ajay Deshpande
2018-10-08 12:01:12 -07:00
committed by zach dwiel
parent 7f00235ed5
commit 0e121c5762
3 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
rl_coach/memories/backend/redis.py
View File

@@ -7,6 +7,7 @@ from kubernetes import client
from rl_coach.memories.backend.memory import MemoryBackend, MemoryBackendParameters from rl_coach.memories.backend.memory import MemoryBackend, MemoryBackendParameters
from rl_coach.core_types import Transition, Episode from rl_coach.core_types import Transition, Episode
from rl_coach.core_types import RunPhase
class RedisPubSubMemoryBackendParameters(MemoryBackendParameters): class RedisPubSubMemoryBackendParameters(MemoryBackendParameters):
@@ -148,7 +149,9 @@ class RedisSub(threading.Thread):
def run(self): def run(self):
for message in self.pubsub.listen(): for message in self.pubsub.listen():
if message and 'data' in message: if message and 'data' in message and self.agent.phase != RunPhase.TEST or self.agent.ap.task_parameters.evaluate_only:
if self.agent.phase == RunPhase.TEST:
print(self.agent.phase)
try: try:
obj = pickle.loads(message['data']) obj = pickle.loads(message['data'])
if type(obj) == Transition: if type(obj) == Transition:

1
rl_coach/training_worker.py
View File

@@ -38,7 +38,6 @@ def training_worker(graph_manager, checkpoint_dir):
graph_manager.phase = core_types.RunPhase.UNDEFINED graph_manager.phase = core_types.RunPhase.UNDEFINED
graph_manager.evaluate(graph_manager.evaluation_steps) graph_manager.evaluate(graph_manager.evaluation_steps)
graph_manager.save_checkpoint() graph_manager.save_checkpoint()
time.sleep(10)
def main(): def main():