some delete

Felixvillas · Jul 7, 2022 · b956be0 · b956be0
1 parent e7dc4a0
commit b956be0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/learn.py b/learn.py
@@ -241,7 +241,7 @@ def dqn_learning(
                 writer.add_scalar(tag=f'starcraft{env_name}_eval/length', scalar_value=mean(eval_step), global_step=t+1)
                 writer.add_scalar(tag=f'starcraft{env_name}_eval/wintag', scalar_value=mean(eval_win), global_step=t+1)
 
-    ### Log progress and keep track of statistics
+    ### log train results
     df = pd.DataFrame({})
     df.insert(loc=0, column='rewards', value=log_rewards)
     df.insert(loc=1, column='steps', value=log_steps)
@@ -267,4 +267,4 @@ def dqn_learning(
     plt.savefig(log_dir + env_name)
 
     writer.close()
-    env.close()
+    env.close()