v0.1.6

robogauge/utils/helpers.py

@@ -58,6 +58,8 @@ def parse_args():
     for param in parameters:
         parser.add_argument(param['name'], **{k: v for k, v in param.items() if k != 'name'})
     args = parser.parse_args()
-    if args.experiment_name is None:
-        args.experiment_name = f"exp"
+    if args.experiment_name is not None:
+        args.experiment_name = f"{args.task_name}_{args.experiment_name}"
+    else:
+        args.experiment_name = args.task_name
     return args

robogauge/utils/logger.py

@@ -104,10 +104,15 @@ class Logger:
             self.logger.addHandler(fh)
         self.info(f"Logs saved at: {path_log_file}")
     
+    def get_data_path(self, robot_name: str, model_name: str, goal_name: str) -> Path:
+        data_path = Path(ROBOGAUGE_LOGS_DIR) / self.experiment_name / 'data' / robot_name / model_name / goal_name / self.tag
+        data_path.mkdir(parents=True, exist_ok=True)
+        return data_path
+    
     def create_tensorboard(self, robot_name: str, model_name: str, goal_name: str):
         if self.writer is not None:
             self.writer.close()
-        data_path = Path(ROBOGAUGE_LOGS_DIR) / self.experiment_name / 'data' / robot_name / model_name / goal_name / self.tag
+        data_path = self.get_data_path(robot_name, model_name, goal_name)
         self.writer = SummaryWriter(str(data_path))
         self.info(f"Tensorboard writer created at: {data_path}")