Add plain text (csv) logging of the success_rate (more readable outsi…

…de of tensorboard)
little-nem · Jun 5, 2019 · b6d63e2 · b6d63e2
1 parent a72fe13
commit b6d63e2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -207,3 +207,5 @@
         print("Testing at episode {}, success rate : {}".format(ep, success_number/TESTING_ROLLOUTS))
         agent.save_model("{}/ckpt_episode_{}".format(directory, ep))
         agent.update_success(success_number/TESTING_ROLLOUTS, ep)
+        with open("csv_log.csv", "a") as csv_log:
+            csv_log.write("{}; {}\n".format(ep, success_number/TESTING_ROLLOUTS))