the error occured when I pretrain the script!  is the released training code corrected?