貓的速度
你正在開發一個新的網路服務,會處理 Instagram 動態來辨識哪些照片裡有貓(別問為什麼——這就是網路)。用來處理資料的程式碼比你預期的還慢,所以你正嘗試調校,讓它跑得更快。給定一張影像 image,你有兩個函式可以處理它:
process_with_numpy(image)process_with_pytorch(image)
你的同事寫了一個情境管理器 timer(),會列印出情境區塊內的程式碼執行花費多久。她建議你用它來看看這兩個選項哪個比較快。請分別計時每個函式,判斷要在你的網路服務中使用哪一個。
本練習屬於課程
Python 函式寫作
練習說明
- 使用
timer()情境管理器來計時process_with_numpy(image)執行所需的時間。 - 使用
timer()情境管理器來計時process_with_pytorch(image)執行所需的時間。
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
image = get_image_from_instagram()
# Time how long process_with_numpy(image) takes to run
____ ____:
print('Numpy version')
process_with_numpy(image)
# Time how long process_with_pytorch(image) takes to run
____ ____:
print('Pytorch version')
process_with_pytorch(image)