Skip to content

Commit

Permalink
Merge pull request #404 from muupan/faster-acer-test
Browse files Browse the repository at this point in the history
Reduce ACER's test cases and maximum timesteps
  • Loading branch information
muupan authored Feb 28, 2019
2 parents 94a0d41 + 689c83b commit 88e1b44
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/agents_tests/test_acer.py
Original file line number Diff line number Diff line change
Expand Up @@ -334,7 +334,7 @@ def loss_func_with_constraint(distrib):
testing.product({
'discrete': [True, False],
't_max': [5],
'use_lstm': [True, False],
'use_lstm': [True],
'episodic': [True, False],
'n_times_replay': [0, 2],
'disable_online_update': [True, False],
Expand All @@ -357,7 +357,7 @@ def test_abc_fast(self):
episodic=self.episodic, steps=10, require_success=False)

def _test_abc(self, t_max, use_lstm, discrete=True, episodic=True,
steps=1000000, require_success=True):
steps=100000, require_success=True):

nproc = 8

Expand Down Expand Up @@ -461,7 +461,7 @@ def phi(x):
beta = 1e-5
if self.n_times_replay == 0 and self.disable_online_update:
# At least one of them must be enabled
self.disable_online_update = False
return
agent = acer.ACER(
model, opt, replay_buffer=replay_buffer,
t_max=t_max, gamma=gamma, beta=beta,
Expand Down

0 comments on commit 88e1b44

Please sign in to comment.