'''
@作者:你遇了我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")