wassname/DeepRL
1
0
Fork 0
mirror of https://github.com/wassname/DeepRL.git synced 2026-09-10 11:40:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refine API

Browse Source
This commit is contained in:
Shangtong Zhang
2018-05-11 09:28:42 -06:00
parent 2ab14b83df
commit d33f85a2c5
2 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
deep_rl/component/task.py
+1 -1
View File
@@ -119,7 +119,7 @@ class Bullet(BaseTask):
return BaseTask.step(self, np.clip(action, -1, 1))
class ProcessTask:
def __init__(self, task_fn, log_dir):
def __init__(self, task_fn, log_dir=None):
self.pipe, worker_pipe = mp.Pipe()
self.worker = ProcessWrapper(worker_pipe, task_fn, log_dir)
self.worker.start()