For pretrain model