Skip to content

Commit 09574d1

Browse files
committed
Rollback acer changes for another commit
1 parent 42e3fa9 commit 09574d1

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

rl_algorithms/acer/agent.py

-3
Original file line numberDiff line numberDiff line change
@@ -93,8 +93,6 @@ def select_action(self, state: np.ndarray) -> Tuple[int, torch.Tensor]:
9393
prob = F.softmax(self.learner.actor_target(state).squeeze(), 0) + 1e-8
9494
action_dist = Categorical(prob)
9595
selected_action = action_dist.sample().item()
96-
if self.is_test:
97-
return selected_action
9896
return selected_action, prob.cpu().numpy()
9997

10098
def step(self, action: int) -> Tuple[np.ndarray, np.float64, bool, dict]:
@@ -159,7 +157,6 @@ def train(self):
159157

160158
if self.i_episode % self.save_period == 0:
161159
self.learner.save_params(self.i_episode)
162-
self.interim_test()
163160

164161
self.env.close()
165162
self.learner.save_params(self.i_episode)

0 commit comments

Comments
 (0)