save checkpoint
add undef.h to advoid nvcc binder
Network learns, output compatiable with Alex's