Open Black3rror opened 10 months ago
@yyoon Could you please check? Thanks!
I have also encountered this problem using TensorFlow version 12.2.1 on my system. Non-optimized conversion works fine with LSTM, but float16 optimization is causing my kernel to crash repeatedly.
Same problem here.
No matter the size of the LSTM model, converting it with float16 optimization runs out of memory.
Code to reproduce the issue The code snippet to reproduce the issue on Google Colab Code: