function w2nn.cleanup_model(model)
   return model:clearState()
end