Model Editing

I have a model which is located in a github-link (https://github.com/khalooei/ALOCC-CVPR2018). This model is being experimented on MNIST and UCSD datasets, but I want this model to be modified to work on my own dataset (fingerprint dataset). The code should be executed on Kaggle.com, so as to enable the used of kaggle's powerful GPU resources. Further details will be given when required. Thanks