Course 5 Week 1 Assignment 3, Error in music_inference_model

The instructions for step 2.D say to use argmax “along the last axis”.
That means you need to include “axis = -1” in the arguments for argmax().

1 Like