Tf.stack is hanging

I have a user who is trying to run some tensorflow code that runs on their laptop but seems to hang on the University’s cluster. The part where it seems to hang is on a tf.stack step. The tensors they are stacking have dimensions (12800, 4) which causes the cluster to hang, but if they reduce it to (1280, 4) it proceeds normally. Would you know of some settting for python/tensorflow that would need to be tweaked on the cluster that would help with this?

Hi @Arenium, This might be due to the computational speed on the cluster node may be less compared to your laptop. Thank You.