Google Professional Machine Learning Engineer Exam - Topic 3 Question 114 Discussion

Actual exam question for Google's Professional Machine Learning Engineer exam

Question #: 114
Topic #: 3

[All Professional Machine Learning Engineer Questions]

You are training an object detection model using a Cloud TPU v2. Training time is taking longer than expected. Based on this simplified trace obtained with a Cloud TPU profile, what action should you take to decrease training time in a cost-efficient way?

AMove from Cloud TPU v2 to Cloud TPU v3 and increase batch size.

BMove from Cloud TPU v2 to 8 NVIDIA V100 GPUs and increase batch size.

CRewrite your input function to resize and reshape the input images.

DRewrite your input function using parallel reads, parallel processing, and prefetch.

Show Suggested Answer

Suggested Answer: D

The trace in the question shows that the training time is taking longer than expected. This is likely due to the input function not being optimized. To decrease training time in a cost-efficient way, the best option is to rewrite the input function using parallel reads, parallel processing, and prefetch. This will allow the model to process the data more efficiently and decrease training time.Reference:

[Cloud TPU Performance Guide]

[Data input pipeline performance guide]

by Rodolfo at Apr 09, 2026, 10:12 AM

Limited Time Offer

25%

2 months ago

I remember we discussed how optimizing the input pipeline can significantly reduce training time, so option D might be the best choice.

upvoted 0 times

...