fix cp of nn to gpu