Merge branch 'master' of github.com:datawhalechina/easy-rl

codes/Sarsa/task0_train.py

@@ -5,7 +5,7 @@ Author: John
 Email: johnjim0816@gmail.com
 Date: 2021-03-11 17:59:16
 LastEditor: John
-LastEditTime: 2021-07-14 17:27:40
+LastEditTime: 2021-07-15 08:52:59
 Discription: 
 Environment: 
 '''
@@ -52,14 +52,15 @@ def train(cfg,env,agent):
         # An episode is an array of (state, action, reward) tuples
         state = env.reset()
         ep_reward = 0
+        action = agent.choose_action(state)
         while True:
         # for t in range(cfg.n_steps):
-            action = agent.choose_action(state)
             next_state, reward, done = env.step(action)
             ep_reward+=reward
             next_action = agent.choose_action(next_state)
             agent.update(state, action, reward, next_state, next_action,done)
             state = next_state
+            action = next_action
             if done:
                 break  
         if ma_rewards: