Add `iter` parameter

Train for `iter` epochs