35 lines
776 B
Python
35 lines
776 B
Python
'''
|
|
@作者:你遇了我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") |