Loading and pre-training on go dataset