python pytorch测试执行时间

目录

pytorch测试执行时间

Python计算程序运行时间

---

pytorch测试执行时间

   start_time = time.time()
   # get predictions
   with torch.no_grad():
       output, _ = model(image)
   torch.cuda.synchronize()
   print('net  time',time.time()-start_time)

注意：

这句代码需要加上，否则测试时间可能不准。

torch.cuda.synchronize()

Python计算程序运行时间

方法1

import datetime
starttime = datetime.datetime.now()

#long running
endtime = datetime.datetime.now()
print (endtime - starttime).seconds

方法 2

import time
import os

if __name__ == '__main__':

    start = time.time()
    run_fun()

    end = time.time()
    print("run time",end - start)

方法3

start = time.clock()

run_fun()

end = time.clock()

print(end-start)

方法1和方法2都包含了其他程序使用CPU的时间，是程序开始到程序结束的运行时间。

方法3算只计算了程序运行的CPU时间