ptrblck:

information


I guess the problem is in image decoding. CPU is always busy in that…
I saw this crop and resize library which I included (https://github.com/longcw/RoIAlign.pytorch 5)
But again my image is 1280x1920x3 so, that’s the block
In my pickle file, I store the image_loc_info,crop_info for a set  of images
Then I use PIL to crop and resize
self.trainloader = torch.utils.data.DataLoader(self.trainset, batch_size=self.batch_size,shuffle=True, num_workers=20,pin_memory=True,collate_fn=self.trainset.collate_fn)
Iterate over the trainloader
inputs_1 = Variable(inputs.cuda(device=self.gpus[0],async=True),volatile=False)
…