Skip to content

Commit

Permalink
minor fix on log and on render
Browse files Browse the repository at this point in the history
  • Loading branch information
goncamateus committed Feb 24, 2021
1 parent 221688a commit 33e825b
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion envs/agents/deepvss/agents/agentDDPGMA.py
Original file line number Diff line number Diff line change
Expand Up @@ -154,7 +154,7 @@ def play(params, net, device, exp_queue, agent_env, test, writer, collected_samp
exp_queue.put(exp)

if test:
agent_env.render()
agent_env.render('human')

states = next_states

Expand Down
2 changes: 1 addition & 1 deletion envs/rc_gym/vss/env_ma/vss_gym_ma.py
Original file line number Diff line number Diff line change
Expand Up @@ -264,7 +264,7 @@ def _calculate_reward_and_done(self):
w_move * move_reward + \
w_energy * energy_penalty

reward[f'robot_{idx}'] = rew
reward[f'robot_{idx}'] += rew
self.reward_shaping_total[f'robot_{idx}']['move'] += w_move * move_reward # noqa
self.reward_shaping_total[f'robot_{idx}']['energy'] += w_energy * energy_penalty # noqa

Expand Down

0 comments on commit 33e825b

Please sign in to comment.