from torchvision.models import resnet34
import numpy as np
import torch


def get_model():
    model = resnet34(pretrained=True)
    num_fc_ftr = model.fc.in_features
    model.fc = torch.nn.Linear(num_fc_ftr, 2)
    return model

e01ca89b58c9ebb4dcd9d1a339408a1813382e5f
<mask token>

<mask token>


def get_model():
    model = resnet34(pretrained=True)
    num_fc_ftr = model.fc.in_features
    model.fc = torch.nn.Linear(num_fc_ftr, 2)
    return model

from torchvision.models import resnet34
import numpy as np
import torch


def get_model():
    model = resnet34(pretrained=True)
    num_fc_ftr = model.fc.in_features
    model.fc = torch.nn.Linear(num_fc_ftr, 2)
    return model
