identified a serious bug in the augmented dataset generation. net works much better now