From baab903a5014e27d0c1f72fcb7b64953d1d542dd Mon Sep 17 00:00:00 2001 From: Dan Hawk Date: Fri, 5 Jul 2019 09:41:11 +0300 Subject: [PATCH] fix tests --- docker/Dockerfile.gpu | 2 +- lib/RLTrader.py | 2 +- requirements.no-gpu.txt | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/Dockerfile.gpu b/docker/Dockerfile.gpu index 3d395b8..cf84ffa 100644 --- a/docker/Dockerfile.gpu +++ b/docker/Dockerfile.gpu @@ -1,7 +1,7 @@ FROM python:3.6.8-jessie -ADD ./requirements.txt /code/ ADD ./requirements.base.txt /code/ +ADD ./requirements.txt /code/ WORKDIR /code diff --git a/lib/RLTrader.py b/lib/RLTrader.py index 97b3ca9..09f2cd9 100644 --- a/lib/RLTrader.py +++ b/lib/RLTrader.py @@ -126,7 +126,7 @@ def optimize_params(self, trial, n_prune_evals_per_trial: int = 2, n_tests_per_e validation_env = SubprocVecEnv([make_env(validation_provider, i) for i in range(1)]) model_params = self.optimize_agent_params(trial) - model = self.Model(self.Policy, train_env, verbose=self.model_verbose, nminibatches=1, + model = self.Model(self.PolicySub, train_env, verbose=self.model_verbose, nminibatches=1, tensorboard_log=self.tensorboard_path, **model_params) last_reward = -np.finfo(np.float16).max diff --git a/requirements.no-gpu.txt b/requirements.no-gpu.txt index cd59df3..72d14c5 100644 --- a/requirements.no-gpu.txt +++ b/requirements.no-gpu.txt @@ -1,2 +1,2 @@ --r requirements.base.yml +-r requirements.base.txt tensorflow