DeepMicrobes Running DeepMicrobes with TensorFlow 2.x

Hello, our GPU requires CUDA 11 and therefore I am stuck with using TensoFlow 2.x. I was trying to lift the code to TF2, but it seems I can't find a replacement for tf.contrib.rnn.CoupledInputForgetGateLSTMCell in the embed_lstm_attention model. Is there by any chance a version of DeepMicrobes that works with TF2? Alternatively, any pointers toward what could be a good replacement for CoupledInputForgetGateLSTMCell would help me a lot.

Sep 19 '22 09:09 Mashin6

Hello, I'm not familiar with TF2, sorry. But I found this: https://discuss.tensorflow.org/t/corresponding-function-in-tfv2-x-x/3405

Sep 19 '22 10:09 MicrobeLab

Thanks, I'll check that out.

Sep 19 '22 11:09 Mashin6

I solved this by changing tf.contrib.rnn.CoupledInputForgetGateLSTMCell to tf.compat.v1.nn.rnn_cell.LSTMCell in models/custom_layers.py

Jun 08 '23 18:06 GiovanniFaldani

Great work, thanks!

Jun 09 '23 02:06 MicrobeLab