Move nn/ to nn_trainer/