I’m training a custom cnn architecture on change detection dataset (http://www.changedetection.net/ 7) .Total number of training images are 35000. It took 7 hours for 1 epoch in lasagne but 15 + hours in pytorch. To load the data, i used DataLoader function present in torch.utils.data. The code is running on a Tesla K20c 5 gb gpu. Batch size of 5 occupies 4.5 gb. num_workers = 4