Update nntrain_gpu.m