pass comment to summary writer

Author: reiniscimurs

robot_nav/models/BPG/BPG.py

@@ -84,7 +84,7 @@ def __init__(
         self.action_dim = action_dim
         self.max_action = max_action
         self.state_dim = state_dim
-        self.writer = SummaryWriter()
+        self.writer = SummaryWriter(comment=model_name)
         self.iter_count = 0
         if load_model:
             self.load(filename=model_name, directory=load_directory)

robot_nav/models/DDPG/DDPG.py

@@ -82,7 +82,7 @@ def __init__(
         self.action_dim = action_dim
         self.max_action = max_action
         self.state_dim = state_dim
-        self.writer = SummaryWriter()
+        self.writer = SummaryWriter(comment=model_name)
         self.iter_count = 0
         if load_model:
             self.load(filename=model_name, directory=load_directory)

robot_nav/models/TD3/TD3.py

@@ -98,7 +98,7 @@ def __init__(
         self.action_dim = action_dim
         self.max_action = max_action
         self.state_dim = state_dim
-        self.writer = SummaryWriter()
+        self.writer = SummaryWriter(comment=model_name)
         self.iter_count = 0
         if load_model:
             self.load(filename=model_name, directory=load_directory)

robot_nav/train.py

@@ -36,15 +36,16 @@ def main(args=None):
     pretraining_iterations = (
         10  # number of training iterations to run during pre-training
     )
-    save_every = 10  # save the model every n training cycles
+    save_every = 5  # save the model every n training cycles
 
-    model = BPG(
+    model = TD3(
         state_dim=state_dim,
         action_dim=action_dim,
         max_action=max_action,
         device=device,
         save_every=save_every,
         load_model=False,
+        model_name="TD3exp1"
     )  # instantiate a model
 
     sim = SIM_ENV()  # instantiate environment