Yep. Mac M1/M2 GPU cannot be used for machine learning if this bug is not fixed.
JerryWangFromNextDoor
Users receive reputation points by contributing quality content to the forums.
Posts include post and replies published on the forums.
Post authors can mark replies as Accepted to indicate successful solutions.
Apple Developer Forums admins can mark replies as Apple Recommended to indicate an approved solution
Post
Replies
Boosts
Views
Activity
I tried to delete the existing session and reload the saved model everytime before prediction. that does not help at all. memory keeps increasing.