Prepare code for Siamese+Imagenet training