main training logic
training complete
testing logic
done