enforce putting iteration numbers in train