fix the issue when iter=warmup