bugfixes for id model training scripts