猫的速度
您正在开发一个新的 Web 服务,用于处理 Instagram 动态并识别哪些图片里有猫(别问为什么——这是互联网)。处理数据的代码比您期望的要慢,因此您正尝试优化以加快运行速度。给定一张图片 image,您有两个函数可以对其进行处理:
process_with_numpy(image)process_with_pytorch(image)
您的同事编写了一个上下文管理器 timer(),它会打印上下文代码块的运行时长。她建议您使用它来比较这两个选项的速度。请分别计时每个函数,以决定在 Web 服务中使用哪一个。
本练习是课程的一部分
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)