unloading the neural net