Commit e8c8eab 1 parent ba8dda7 commit e8c8eab Copy full SHA for e8c8eab
File tree 1 file changed +10
-12
lines changed
1 file changed +10
-12
lines changed Original file line number Diff line number Diff line change
1
+ import multiprocessing
1
2
import numpy as np
2
3
3
- import multiprocessing
4
4
from lib .RLTrader import RLTrader
5
5
6
6
np .warnings .filterwarnings ('ignore' )
@@ -13,19 +13,17 @@ def optimize_code(params):
13
13
14
14
if __name__ == '__main__' :
15
15
n_process = multiprocessing .cpu_count ()
16
- params = {'n_cpu ' : n_process }
16
+ params = {'n_envs ' : n_process }
17
17
18
- # processes = []
19
- # for i in range(n_process):
20
- # processes.append(multiprocessing.Process(target=optimize_code, args=(params,)))
18
+ processes = []
19
+ for i in range (n_process ):
20
+ processes .append (multiprocessing .Process (target = optimize_code , args = (params ,)))
21
21
22
- # for p in processes:
23
- # p.start()
22
+ for p in processes :
23
+ p .start ()
24
24
25
- # for p in processes:
26
- # p.join()
25
+ for p in processes :
26
+ p .join ()
27
27
28
28
trader = RLTrader (** params )
29
- # trader.train(test_trained_model=True, render_trained_model=True)
30
-
31
- trader .test (model_epoch = 10 )
29
+ trader .train (test_trained_model = True , render_trained_model = True )
You can’t perform that action at this time.
0 commit comments