More fixes to model traninig scripts