完成模型日志记录模块

train/utils/tensorborad_utils.py

@@ -0,0 +1,35 @@
+'''
+@作者:你遇了我321640253@qq.com
+@文件:tensorborad_utils.py
+@创建时间:2023 11 20
+
+
+'''
+
+import time
+import os
+from torch.utils.tensorboard import SummaryWriter
+
+class ModelLog():
+
+    def __init__(self,logdir:str):
+        self.timestr = self.getTimeStr()
+        #获取日志路径
+        logdir = os.path.join(logdir,self.timestr)
+
+        #创建日志
+        self.Write = SummaryWriter(log_dir=logdir)
+
+    def getTimeStr(self):
+        '''
+        :description 获取当前时间
+        :author  你遇了我
+        :param 
+        :return 
+        '''
+        _time = time.gmtime()
+   
+        return f"{_time.tm_year}-{_time.tm_mon}-{_time.tm_mday}-{_time.tm_hour+8}-{_time.tm_min}-{_time.tm_sec}"
+
+if __name__ == "__main__":
+    ModelLog("ModelLog")