text_plain_output_1.png image_output_1.png
import torch.nn as nn

import torch.nn as nn

import torchvision.models as models

model_v = models.resnet18()
model_c = models.resnet18()

model_c.fc = nn.Linear(512, 10, bias=True)
model_v.fc = nn.Linear(512, 10, bias=True)
print(model_c)