Update nntrain_gpu.m

save best network