UW model training