ND data loading + training debugged