dog_breed_classification_arch.py

from torchvision.models.resnet import ResNet, Bottleneck
import torch.nn as nn

class ImageClassifier(ResNet):
    def __init__(self):
        super(ImageClassifier, self).__init__(Bottleneck, [3, 4, 6, 3])
        self.fc = nn.Linear(self.fc.in_features, 133)