fix for nn tester and add training time