Solved by albanD in post #2 
              
              
                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?