Skip to content

Commit

Permalink
FIx eval step iteration
Browse files Browse the repository at this point in the history
  • Loading branch information
Pdbz199 committed Aug 12, 2023
1 parent 54a5e78 commit 0929fd6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion final/control/policies/soft_actor_koopman_critic/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -181,6 +181,7 @@
)
)

eval_steps = 0
if i_episode % args.eval_frequency == 0:
if len(memory) > args.batch_size:
agent.save_checkpoint(args.env_name)
Expand All @@ -192,7 +193,6 @@
sac_avg_reward = 0
lqr_avg_reward = 0
num_eval_episodes = args.eval_episodes
eval_steps = 0

for _ in range(num_eval_episodes):
# initial_state = np.array([1, 1, 1])
Expand Down

0 comments on commit 0929fd6

Please sign in to comment.