I am still experiencing these issues with relatively simple models catastrophically failing to converge on Apple Silicon machines when using the GPU. When running on the CPU, I get better performance and good convergence. The only change is switching from GPU to CPU.
I am using tensorflow-metal v1.1.0, tensorflow v2.16.2, tensorflow-macos v2.16.2. If anyone has found a solution, or even a temporary work-around that yields good performance on the GPU, please advise.
Topic:
Machine Learning & AI
SubTopic:
General
Tags: