DeepMicrobes icon indicating copy to clipboard operation
DeepMicrobes copied to clipboard

Running DeepMicrobes with TensorFlow 2.x

Open Mashin6 opened this issue 1 year ago • 4 comments

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.

Mashin6 avatar 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

MicrobeLab avatar Sep 19 '22 10:09 MicrobeLab

Thanks, I'll check that out.

Mashin6 avatar Sep 19 '22 11:09 Mashin6