bug fix in allowing different df for pretraining