Hi,
Have you tried to reduce the embedding size or replace Adam with SGD?
The Adam optimizers will allocate a couple buffers with the same size as the weights to store all the statistics it needs. So that might be too much for your machine?