Fixed the declaration of the SdA in the pretraining script.