python代码运行时间信息记录

最近训练模型经常要记录运行时间相关的信息。这里简单记录一下，方便之后直接使用。

1. time模块

   使用 time 模块来记录代码的运行时间，参考代码如下：

   import time
   
   # 记录开始时间
   start_time = time.time()
   
   # --------------------------
   # 这里放你要测试运行时间的代码
   # 示例：模拟一段耗时操作
   total = 0
   for i in range(1000000):
       total += i
   # --------------------------
   
   # 记录结束时间
   end_time = time.time()
   
   # 计算并打印运行时间
   run_time = end_time - start_time
   print(f"代码运行时间: {run_time:.6f} 秒")
   # 也可以按毫秒显示，更直观
   # print(f"代码运行时间: {run_time * 1000:.2f} 毫秒")

   输出结果：
   

2. timeit模块

   如果想得到更精确的运行时间测试，可以使用timeit模块，参考代码如下：

   import timeit
   
   # 定义要测试的代码（可以是字符串或函数）
   def test_code():
       total = 0
       for i in range(1000000):
           total += i
   
   # 执行测试：number表示运行次数，默认会自动选择合适的次数
   # stmt：要执行的代码/函数，setup：执行前的初始化代码
   run_time = timeit.timeit(stmt=test_code, setup="pass", number=100)
   # 计算单次运行时间
   avg_time = run_time / 100
   print(f"代码运行100次总时间: {run_time:.6f} 秒")
   print(f"代码单次平均运行时间: {avg_time:.6f} 秒")

   输出结果：
   

3. datetime模块

   获取格式化的当前时间，可以使用datetime模块实现，以便保存训练模型的时间点。

   import datetime
   
   # 获取当前本地时间，并格式化为 年-月-日 时:分:秒
   current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
   print(f"代码当前运行时间: {current_time}")
   
   # 更详细的格式（包含毫秒）
   current_time_ms = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")[:-3]  # 保留3位毫秒
   print(f"代码当前运行时间（含毫秒）: {current_time_ms}")

   运行结果：