Relative Content

Tag Archive for pythonnumpytorchconvolution

Optimizing Convolution Layer on GPU with Torch Tensors

Problem With the below toy script, I am getting worse results on the GPU than the CPU. I am fairly new to GPU programming, so I’m not sure how to even debug. From my limited research, convolution is something that should be able to be greatly sped up by the GPU so I’m guessing I’m […]