switch multiproc thread count to rl_config definition
This commit is contained in:
parent
b26ed7dea4
commit
b708134c1a
@ -60,7 +60,7 @@ class ReinforcementLearner_multiproc(BaseReinforcementLearningModel):
|
|||||||
# environments
|
# environments
|
||||||
if not self.train_env:
|
if not self.train_env:
|
||||||
env_id = "train_env"
|
env_id = "train_env"
|
||||||
num_cpu = int(self.freqai_info["data_kitchen_thread_count"] / 2)
|
num_cpu = int(self.freqai_info["rl_config"]["thread_count"] / 2)
|
||||||
self.train_env = SubprocVecEnv([make_env(env_id, i, 1, train_df, prices_train,
|
self.train_env = SubprocVecEnv([make_env(env_id, i, 1, train_df, prices_train,
|
||||||
self.reward_params, self.CONV_WIDTH,
|
self.reward_params, self.CONV_WIDTH,
|
||||||
config=self.config) for i
|
config=self.config) for i
|
||||||
|
Loading…
Reference in New Issue
Block a user