todo
driver refactoring
test train
support for specific neural network count