Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
dw-yejing committed Dec 5, 2024
1 parent cb95675 commit 8ad15db
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion gym/trainpy.py → gym/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ def train(env, model, optimizer, num_episodes):
print(f"Episode {episode}: Total Reward = {total_reward}; loss: {loss.item()}")
if total_reward > checkpoint['best_reward']:
checkpoint['best_reward'] = total_reward
save_ckpt(model, "runs", optimizer, total_reward)
save_ckpt(model, "out", optimizer, total_reward)

if __name__ == "__main__":
# 创建CartPole环境
Expand Down

0 comments on commit 8ad15db

Please sign in to comment.