version used for large scale training