Setup Wandb

marload · Mar 25, 2020 · a174828 · a174828
1 parent 9883613
commit a174828
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/DuelingDQN/Dueling_DQN_Discrete.py → DuelingDQN/DuelingDQN_Discrete.py b/DuelingDQN/Dueling_DQN_Discrete.py → 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():