add pretrain script