juneoh

import torch.nn as nn
from torchvision.models import alexnet

model = alexnet(pretrained=True, num_classes=10)
model.features[0] = nn.Conv2d(1, 64, kernel_size=(11, 11), stride=(4, 4), padding=(2, 2))