mrdbourke / tensorflow-deep-learning

All course materials for the Zero to Mastery Deep Learning with TensorFlow course.
https://dbourke.link/ZTMTFcourse
MIT License
5.29k stars 2.6k forks source link

Fixing the range of randint #561

Closed rohanbobby01 closed 1 year ago

rohanbobby01 commented 1 year ago

To prevent IndexError

rohanbobby01 commented 1 year ago

To view an image from a batch, random.randint(0, 32) was used, but it will give an index error if it returns 32, so changed it to random.randint(0, 31)

mrdbourke commented 1 year ago

Good catch! Thank you :)