Prediction not working, received error 'cuda could not initialize prediction'

[brianmanderson]

created model seemed to not be able to predict, turning lambdas into standard Layers appeared to fix the issue, pull request in #142