Я пытаюсь выполнить регрессию в модели преобразователей зрения, но не могу заменить последний уровень классификации слоем регрессии
Я получаю эту ошибку при попытке инициализировать модель< /p>
class RegressionViT(nn.Module):
def __init__(self, in_features=224 * 224 * 3, num_classes=1, pretrained=True):
super(RegressionViT, self).__init__()
self.vit_b_16 = vit_b_16(pretrained=pretrained) # Load pre-trained weights
# Replace the final classification layer with a regression head
self.regressor = nn.Linear(self.vit_b_16.heads.in_features, num_classes)
def forward(self, x):
x = self.vit_b_16(x)
x = self.regressor(x)
return x
Подробнее здесь: https://stackoverflow.com/questions/782 ... mers-model
