From a174828f21e9484e42acf0325298fe0e34c8a967 Mon Sep 17 00:00:00 2001
From: marload <rladhkstn8@gmail.com>
Date: Wed, 25 Mar 2020 13:10:03 +0900
Subject: [PATCH] Setup Wandb

---
 .../{Dueling_DQN_Discrete.py => DuelingDQN_Discrete.py}       | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)
 rename DuelingDQN/{Dueling_DQN_Discrete.py => DuelingDQN_Discrete.py} (97%)

diff --git a/DuelingDQN/Dueling_DQN_Discrete.py b/DuelingDQN/DuelingDQN_Discrete.py
similarity index 97%
rename from DuelingDQN/Dueling_DQN_Discrete.py
rename to DuelingDQN/DuelingDQN_Discrete.py
index 9ac4a12..4cef6b5 100644
--- a/DuelingDQN/Dueling_DQN_Discrete.py
+++ b/DuelingDQN/DuelingDQN_Discrete.py
@@ -10,7 +10,7 @@
 import random
 
 tf.keras.backend.set_floatx('float64')
-# wandb.init(name='DuelingDQN', project="deep-rl-tf2")
+wandb.init(name='DuelingDQN', project="deep-rl-tf2")
 
 parser = argparse.ArgumentParser()
 parser.add_argument('--gamma', type=float, default=0.95)
@@ -117,7 +117,7 @@ def train(self, max_episodes=1000):
                 state = next_state
             self.target_update()
             print('EP{} EpisodeReward={}'.format(ep, total_reward))
-            # wandb.log({'Reward': total_reward})
+            wandb.log({'Reward': total_reward})
 
 
 def main():