Minor Code cleanup in utils file

Made change to cleaning code, modified number of warmpu step, getting eval samples from validation split

Updated code to work with streaming version