Sample code for creating a modified dataset class in torch that can be used for training.
Primary LanguagePython