before stageful training data fixing
tmp
fixed stagefule training data