def __init__(self, hidden_dim, net_type, pretrained):
    super(SketchCNN, self).__init__()
    if (net_type == 'resnet18'):
        self.encoder = models.resnet18(pretrained=pretrained)
        self.encoder.fc = nn.Linear(self.encoder.fc.in_features, hidden_dim)
    elif (net_type == 'resnet50'):
        self.encoder = models.resnet50(pretrained=pretrained)
        self.encoder.fc = nn.Linear(self.encoder.fc.in_features, hidden_dim)
    elif (net_type == 'tcnet'):
        pass
    elif (net_type == 'sketchanet'):
        pass
def __init__(self, hidden_dim, net_type, pretrained):
    super(SketchCNN, self).__init__()
    if (net_type == 'resnet18'):
        self.encoder = models.resnet18(pretrained=pretrained)
        self.encoder.fc = nn.Linear(self.encoder.fc.in_features, hidden_dim)
    elif (net_type == 'resnet50'):
        self.encoder = models.resnet50(pretrained=pretrained)
        self.encoder.fc = nn.Linear(self.encoder.fc.in_features, hidden_dim)
    elif (net_type == 'tcnet'):
        pass
    elif (net_type == 'sketchanet'):
        pass