Solved by ptrblck in post #2 
              
              
                Try to reduce the batch size and check which batch size would fit and how much memory is used. 
Your GPUs might have not enough memory for the code you are using or are you using exactly the setup the authors were also using?