未验证 提交 1f8a830b 编写于 作者: T Tingquan Gao 提交者: GitHub

fix model name (#2908)

* fix model name

* fix: bug when distillation
上级 77aa0706
......@@ -30,7 +30,7 @@ class MLP(nn.Layer):
class Aesthetic_Score_Predictor(nn.Layer):
def __init__(self):
super().__init__()
self.model = CLIP_large_patch14_224()
self.model = CLIP_vit_large_patch14_224()
self.fc_head = nn.Linear(1024, 768, bias_attr=False)
self.mlp = MLP(768)
......
......@@ -57,6 +57,7 @@ class CELoss(nn.Layer):
loss = loss.sum()
else:
if label.shape[-1] == x.shape[-1]:
label = F.softmax(label, axis=-1)
soft_label = True
else:
soft_label = False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册