I have the same problem.
Apple M2
macOS 13.4.1(c)
tensorflow 2.14.0
tensorflow-metal 1.1.0
Memory seems to grow by 10GB per iteration even if I release everything for each iteration and call gc.collect() afterwards, using this model:
https://github.com/google-research/frame-interpolation
The memory is released when the process is terminated, but this makes it impossible to run large models using tensor flow-metal. Everything works fine when using the CPU.
Ola.
Ola_F_G
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