cyy/RoboGauge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

v0.1.6

Browse Source
This commit is contained in:
wty-yy
2025-12-03 09:43:16 +08:00
parent ad907089a8
commit 9f7db42f9f
12 changed files with 179 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
robogauge/utils/logger.py
View File

@@ -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}")