max_loop for pretrain altered.