Solved by tom in post #2 
              
              
                Your net only takes cuda floats as input, so your data preparation does not match the expectation of later processing stages when it casts to long. The solution to that depends on what you want to achieve, leaving as float or one-hot-encoding your integer in a real vector are common things to do. 
B…