switch multiproc thread count to rl_config definition

This commit is contained in:
robcaulk 2022-08-23 15:06:57 +02:00
parent b26ed7dea4
commit b708134c1a

View File

@ -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