optimize comments

SwanHubX · Jan 20, 2025 · 9c81230 · 9c81230
1 parent 2a0b105
commit 9c81230
Showing 1 changed file with 6 additions and 14 deletions.
diff --git a/swanlab/sync/tensorboard.py b/swanlab/sync/tensorboard.py
@@ -6,19 +6,15 @@ def sync_tensorboardX():
     
     from tensorboardX import SummaryWriter
     import numpy as np
-
-    # 创建一个SummaryWriter对象，指定日志目录
+    import swanlab
+    
+    swanlab.sync_tensorboardX()
     writer = SummaryWriter('runs/example')
 
-    # 生成一些示例数据
     for i in range(100):
-        # 生成一个随机的标量值
         scalar_value = np.random.rand()
-        
-        # 记录标量值
         writer.add_scalar('random_scalar', scalar_value, i)
 
-    # 关闭SummaryWriter
     writer.close()
     """
     try:
@@ -68,19 +64,15 @@ def sync_tensorboard_torch():
     
     from torch.utils.tensorboard import SummaryWriter
     import numpy as np
-
-    # 创建一个SummaryWriter对象，指定日志目录
+    import swanlab
+    
+    swanlab.sync_tensorboard_torch()
     writer = SummaryWriter('runs/example')
 
-    # 生成一些示例数据
     for i in range(100):
-        # 生成一个随机的标量值
         scalar_value = np.random.rand()
-        
-        # 记录标量值
         writer.add_scalar('random_scalar', scalar_value, i)
 
-    # 关闭SummaryWriter
     writer.close()
     """
     try: