Spaces:
Build error
Build error
File size: 225 Bytes
a7f04f4 | 1 2 3 4 5 6 7 8 9 10 | import torch
import torch.nn as nn
import torchvision.models as models
weights = models.ResNet50_Weights.DEFAULT
resnet_model = models.resnet50(weights=weights)
resnet_model.fc = nn.Linear(resnet_model.fc.in_features , 3)
|