Hi,
I’m new to pytorch and for a student project I’d like to setup a simple convolution for an image which is 26 megapixel. The below code fails with “RuntimeError: could not create a primitive”. When I decrease the size of the ´torch.randn´ the conv2d works as expected. Where is my error? The code is from a larger project SESF_Net on github. I’m just trying to learn and understand.
import os
import torch
import torch.nn as nn
import torch.nn.functional as f
r_shift_kernel = torch.FloatTensor([[0, 0, 0], [1, 0, 0], [0, 0, 0]]).reshape((1, 1, 3, 3)).repeat(64, 1, 1, 1)
f1 = torch.randn(1,64,4160,6240)
f1_r_shift = f.conv2d(f1, r_shift_kernel, padding=1, groups=64)

Thanks and best wishes.