efficientnet + amp BCE
Efficientnet 모델 구성 from torch import nn import timm import torch class EffNet(nn.Module): def __init__(self, backbone, n_out, is_sigmoid): super(EffNet, self).__init__() self.model = timm.create_model(model_name=backbone, pretrained=True) self.model.classifier = nn.Linear(self.model.classifier.in_features, n_out) self.is_sigmoid = is_sigmoid def forward(self, x): x = self.model(x) if self.is_sig..
AI
2023. 7. 6. 09:25
