from torchvision.models import resnet50
net = resnet50()
net.conv1.in_channels = 9
print(net)
1 Like
from torchvision.models import resnet50
net = resnet50()
net.conv1.in_channels = 9
print(net)