albanD:

Could it be that the kernel scales very badly with the input size?


Do you know how to use VOC2007 dataset instead of the CIFAR-10 dataset (which is only of size 32 x 32) on the CUDA kernel itself ?
torchvision.dataset does not have VOC2007