Skip to content

Commit

Permalink
rmse fix for ppo and sac
Browse files Browse the repository at this point in the history
  • Loading branch information
svsawant committed Aug 21, 2024
1 parent 54fb070 commit bdf26c4
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion safe_control_gym/controllers/ppo/ppo.py
Original file line number Diff line number Diff line change
Expand Up @@ -252,7 +252,7 @@ def run(self,
if done:
assert 'episode' in info
ep_rmse_mean.append(np.array(mse).mean()**0.5)
ep_rmse_std.append(np.array(mse).std()**0.5)
ep_rmse_std.append(np.array(mse).std())
mse = []
ep_returns.append(info['episode']['r'])
ep_lengths.append(info['episode']['l'])
Expand Down
2 changes: 1 addition & 1 deletion safe_control_gym/controllers/sac/sac.py
Original file line number Diff line number Diff line change
Expand Up @@ -251,7 +251,7 @@ def run(self, env=None, render=False, n_episodes=10, verbose=False, **kwargs):
if done:
assert 'episode' in info
ep_rmse_mean.append(np.array(mse).mean()**0.5)
ep_rmse_std.append(np.array(mse).std()**0.5)
ep_rmse_std.append(np.array(mse).std())
mse = []
ep_returns.append(info['episode']['r'])
ep_lengths.append(info['episode']['l'])
Expand Down

0 comments on commit bdf26c4

Please sign in to comment.