Update deep q learning.py

wybin4 · Apr 25, 2022 · 70dd450 · 70dd450
1 parent c64af83
commit 70dd450
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deep q learning.py b/deep q learning.py
@@ -172,7 +172,7 @@ def main():
         print(f"on #{episode}, epsilon is {epsilon}")
         print(f"ep mean: {np.mean(total_training_rewards)}")        
         all_episodes_reward.append(total_training_rewards)
-        q_model.save(f'models/{MODEL_NAME}__{np.mean(all_episodes_reward[-25:])}mean__{int(time.time())}.model')
+    q_model.save(f'models/{MODEL_NAME}__{np.mean(all_episodes_reward[-25:])}mean__{int(time.time())}.model')
     return all_episodes_reward
 
 if __name__ == '__main__':