The speed of cats
You're working on a new web service that processes Instagram feeds to identify which pictures contain cats (don't ask why -- it's the internet). The code that processes the data is slower than you would like it to be, so you are working on tuning it up to run faster. Given an image, image
, you have two functions that can process it:
process_with_numpy(image)
process_with_pytorch(image)
Your colleague wrote a context manager, timer()
, that will print out how long the code inside the context block takes to run. She is suggesting you use it to see which of the two options is faster. Time each function to determine which one to use in your web service.
This exercise is part of the course
Writing Functions in Python
Exercise instructions
- Use the
timer()
context manager to time how longprocess_with_numpy(image)
takes to run. - Use the
timer()
context manager to time how longprocess_with_pytorch(image)
takes to run.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
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)