I am trying to use TPU in colab to speed up the training process of LSTM. But I am keep getting the error: Unable to find the relevant tensor remote_handle: Op ID: 529, Output num: 0
The code and data link are shared below, can anyone please help me to fix this error? Much appreciated :-)
colab code: https://colab.research.google.com/drive/1t1kDuehLNjVhrtmCXdeSAbPcI5Vht90P?usp=sharing data file（OHLC and indicators，139 columns in total）：https://drive.google.com/file/d/180Ai703VHdICN1VtQl4zWkZw0vHNMoBm/view?usp=sharing