How can I train the student by other datasets?

Thanks for your great work!
If I want to use my dataset to train the student, how can I generate _trainset_similarities_teacher.pickle_ like you did?
Many thanks!