Added code to allow training in the first 150K iterations